generate_urd.py

from langchain.chains import LLMChain
from langchain.chat_models import ChatOpenAI
from langchain.prompts import SystemMessagePromptTemplate, HumanMessagePromptTemplate, ChatPromptTemplate


def generate_user_requirement_document(openai_api_key, answers):
    system_message = """Given the information by the user, generate a User Requirement Document (URD). The URD should \
have a clear problem statement, goals, objectives to accomplish, features, and a conclusion paragraph. Reproduce the \
features part as it is in the response

Generate the features response in the following format:
Feature Name:
Feature Prerequisites:
Feature Description:

Sub-feature Name:
Sub-feature Prerequisites:
Sub-feature Description:

# Generate the response document in the markdown format 

Must generate the title with the topic name
"""

    human_message = """
Problem: {answer1}
Platform: {platform}
Goal: {answer2}
Features: {answer3}
Users: {answer4}
"""

    system_message_prompt = SystemMessagePromptTemplate.from_template(system_message)
    human_message_prompt = HumanMessagePromptTemplate.from_template(human_message)

    chat_prompt = ChatPromptTemplate.from_messages(
        [
            system_message_prompt,
            human_message_prompt
        ]
    )

    chat_model = ChatOpenAI(openai_api_key=openai_api_key, temperature=0.6, model="gpt-3.5-turbo-16k")

    llm_chain = LLMChain(prompt=chat_prompt, llm=chat_model, verbose=True)
    response = llm_chain.run(
        answer1=answers["answer1"],
        platform=answers["platform"],
        answer2=answers["answer2"],
        answer3=answers["answer3"],
        answer4=answers["answer4"],
    )

    return response