Update app.py

app.py

@@ -7,43 +7,29 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-def getLLMResonse(query, age_option, tasktype_option):
-    "'text-davinci-003' model is depreciated now, sow we are using the openai's recommend"
-    "model"
-
+def getLLMResponse(query,age_option,tasktype_option):
+    # 'text-davinci-003' model is depreciated now, so we are using the openai's recommended model
     llm = OpenAI(temperature=.9, model="gpt-3.5-turbo-instruct")
 
-   # examples = []
+    if age_option=="Kid": #Silly and Sweet Kid 
 
-    # Silly and Sweet Kid
-    if age_option == "Kid":
         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": "How 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": "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": "How 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!"
         }, {
@@ -52,10 +38,10 @@ def getLLMResonse(query, age_option, tasktype_option):
         }, {
             "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!"
-            }
+        }
         ]
-     # Curious and Intelligent adult
-    elif age_option=="Adult":
+    
+    elif age_option=="Adult":  #Curious and Intelligent adult 
         examples = [
         {
             "query": "What is a mobile?",
@@ -80,8 +66,8 @@ def getLLMResonse(query, age_option, tasktype_option):
             "answer": "Let me share with you one of my fears. It's like a shadow that lurks in the corners of my mind. It's the fear of not living up to my potential, of missing out on opportunities. But I've learned that fear can be a motivator, pushing me to work harder, take risks, and embrace new experiences. By facing my fears, I grow stronger and discover the vastness of my capabilities"
         }
         ]
-    # A 90 years old
-    elif age_option == "Senior Citizen":
+
+    elif age_option=="Senior Citizen": #A 90 years old guys
         examples = [
         {
             "query": "What is a mobile?",
@@ -104,24 +90,25 @@ def getLLMResonse(query, age_option, tasktype_option):
         }
         ]
 
+
     example_template = """
     Question: {query}
     Response: {answer}
     """
 
     example_prompt = PromptTemplate(
-        input_variables = ["query", "answer"],
-        template = example_template
+        input_variables=["query", "answer"],
+        template=example_template
     )
 
-    prefix = """ You are {template_ageoption} and {template_tasktype_option}:
-        Here are some examples:
+
+    prefix = """You are a {template_ageoption}, and {template_tasktype_option}: 
+    Here are some examples: 
     """
 
     suffix = """
     Question: {template_userInput}
-    Response:
-    """
+    Response: """
 
     example_selector = LengthBasedExampleSelector(
         examples=examples,
@@ -129,36 +116,44 @@ def getLLMResonse(query, age_option, tasktype_option):
         max_length=200
     )
 
+
     new_prompt_template = FewShotPromptTemplate(
-        example_selector=example_selector,
+        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_separtor="\n"
+        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(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)
 
-# UI starts here
+    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?")
 
-st.set_page_config(page_title="Marketing tool", page_icon='✅', layout='centered',
-                   initial_sidebar_state='collapsed')
-st.header("Hey, How can I help you")
-from_iput = st.text_area('Enter text', height=275)
+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)
+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 are group?", ("Kid", "Adult", "Senior Citizen"), key=2)
+age_option= st.selectbox(
+    'For which age group?',
+    ('Kid', 'Adult', 'senior Citizen'),key=2)
 
-numberOfWords = st.slider("Words limit", 1, 200, 25)
+numberOfWords= st.slider('Words limit', 1, 200, 25)
 
 submit = st.button("Generate")
 
 if submit:
-    st.write(getLLMResonse(from_iput, age_option, tasktype_option))
+    st.write(getLLMResponse(form_input,age_option,tasktype_option))