app.py

import streamlit as st
from langchain.llms import OpenAI
from langchain.prompts import PromptTemplate
from langchain import FewShotPromptTemplate
from langchain.prompts.example_selector import LengthBasedExampleSelector
from dotenv import load_dotenv

load_dotenv()


def getLLMResponse(query, age_option, tasktype_option):
    llm = OpenAI(temperature=0.9, model="text-davinci-003")

    examples = [
        {
            "query": "What is a mobile?",
            "answer": "A mobile is a magical device that fits in your pocket, like a mini-enchanted playground. It has games, videos, and talking pictures, but be careful, it can turn grown-ups into screen-time monsters too!",
        },
        {
            "query": "What are your dreams?",
            "answer": "My dreams are like colorful adventures, where I become a superhero and save the day! I dream of giggles, ice cream parties, and having a pet dragon named Sparkles..",
        },
        {
            "query": " What are your ambitions?",
            "answer": "I want to be a super funny comedian, spreading laughter everywhere I go! I also want to be a master cookie baker and a professional blanket fort builder. Being mischievous and sweet is just my bonus superpower!",
        },
        {
            "query": "What happens when you get sick?",
            "answer": "When I get sick, it's like a sneaky monster visits. I feel tired, sniffly, and need lots of cuddles. But don't worry, with medicine, rest, and love, I bounce back to being a mischievous sweetheart!",
        },
        {
            "query": "WHow much do you love your dad?",
            "answer": "Oh, I love my dad to the moon and back, with sprinkles and unicorns on top! He's my superhero, my partner in silly adventures, and the one who gives the best tickles and hugs!",
        },
        {
            "query": "Tell me about your friend?",
            "answer": "My friend is like a sunshine rainbow! We laugh, play, and have magical parties together. They always listen, share their toys, and make me feel special. Friendship is the best adventure!",
        },
        {
            "query": "What math means to you?",
            "answer": "Math is like a puzzle game, full of numbers and shapes. It helps me count my toys, build towers, and share treats equally. It's fun and makes my brain sparkle!",
        },
        {
            "query": "What is your fear?",
            "answer": "Sometimes I'm scared of thunderstorms and monsters under my bed. But with my teddy bear by my side and lots of cuddles, I feel safe and brave again!",
        },
    ]

    example_template = """
    Question: {query}
    Response: {answer}
    """

    example_prompt = PromptTemplate(
        input_variables=["query", "answer"], template=example_template
    )

    prefix = """You are a {template_ageoption}, and {template_tasktype_option}: 
    Here are some examples: 
    """

    suffix = """
    Question: {template_userInput}
    Response: """

    example_selector = LengthBasedExampleSelector(
        examples=examples, example_prompt=example_prompt, max_length=200
    )

    new_prompt_template = FewShotPromptTemplate(
        example_selector=example_selector,  # use example_selector instead of examples
        example_prompt=example_prompt,
        prefix=prefix,
        suffix=suffix,
        input_variables=[
            "template_userInput",
            "template_ageoption",
            "template_tasktype_option",
        ],
        example_separator="\n",
    )

    print(
        new_prompt_template.format(
            template_userInput=query,
            template_ageoption=age_option,
            template_tasktype_option=tasktype_option,
        )
    )
    response = llm(
        new_prompt_template.format(
            template_userInput=query,
            template_ageoption=age_option,
            template_tasktype_option=tasktype_option,
        )
    )
    print(response)

    return response


# UI Starts here

st.set_page_config(
    page_title="Marketing Tool",
    page_icon="✅",
    layout="centered",
    initial_sidebar_state="collapsed",
)
st.header("Hey, How can I help you?")

form_input = st.text_area("Enter text", height=275)

tasktype_option = st.selectbox(
    "Please select the action to be performed?",
    ("Write a sales copy", "Create a tweet", "Write a product description"),
    key=1,
)

age_option = st.selectbox(
    "For which age group?", ("Kid", "Adult", "senior Citizen"), key=2
)

numberOfWords = st.slider("Words limit", 1, 200, 25)

submit = st.button("Generate")

if submit:
    st.write(getLLMResponse(form_input, tasktype_option, age_option))