Update app.py

app.py

@@ -66,17 +66,47 @@ with col2:
 with col1:
     st.image("taylor_1.jpg")
 
-if st.button("Generate Affirmation", key="generate_affirmation"):
-    # prompt = """My name is aditi, i have 2 pets named Jamun and Misri and i do not like going to gym because i expect results too quickly. 
-    # In few days i see no gains and get irritated and give up. This time around i have decided to take it slow and 
-    # do it more for feeling better, reduce resting heart rate, overall feeling fitter little bit everyday.
-    # Your job is to give me positive affirmations like taylor swift would talk to me in the theme i described of taking it slow and doing it more for happiness and feeling fit.
-    # Try to include taylor swift song lyrics in few places not forcefully but relevant to the overall content. 
-    # Important is i am giving all this as context, i don't want output to use any of the words i have written so that it feel fake or forceful.
-    # The structure is basically conversational like taylor swift talking to me about my life and pets and not bullet points.
-    # """
+context = st.text_input("Tell taylor about your day for context")
+
+groq_api_key = os.getenv['groq_api_token']
+
+from groq import Groq
+
+client = Groq(
+    api_key=groq_api_key,
+)
 
+def llama3_8b(prompt, temperature=0.0, input_print=True):
+  chat_completion = client.chat.completions.create(
+      messages=[
+          {
+              "role": "user",
+              "content": prompt,
+          }
+      ],
+      model="llama3-8b-8192",
+      temperature=temperature,
+  )
 
+  return (chat_completion.choices[0].message.content)
+
+def llama3_70b(prompt, temperature=0.0, input_print=True):
+  chat_completion = client.chat.completions.create(
+      messages=[
+          {
+              "role": "user",
+              "content": prompt,
+          }
+      ],
+      model="llama3-70b-8192",
+      temperature=temperature,
+  )
+
+  return (chat_completion.choices[0].message.content)
+
+
+
+if st.button("Generate Affirmation", key="generate_affirmation"):
 
     def get_astrology_sign(birth_date):
         # Zodiac signs divided by the date ranges
@@ -123,25 +153,26 @@ if st.button("Generate Affirmation", key="generate_affirmation"):
     In few days i see no gains and get irritated and give up. This time around i have decided to take it slow and \n
     do it more for feeling better, reduce resting heart rate, overall feeling fitter little bit everyday.\n
     Here\'s my horoscope information for Today {}\n\n
-    Taking all this above context , Your job is to give me positive affirmations like taylor swift would talk to me in the theme of starting with fewn lines about my horoscope and then talk about my fitness goals.\n
+    I have also described how my day has been {}\n\n 
+    Taking all this above context , Your job is to give me positive affirmations like taylor swift would talk to me in the theme of starting with few lines about my life in general and then talk about my fitness goals.\n
     Try to include taylor swift song lyrics in few places not forcefully but relevant to the overall content. \n
     I don\'t want output to use exactly these words i have written so that it does not feel fake or forceful.\n
-    The structure should be conversational like taylor swift talking to me about my life and pets and not bullet points or third person.'''.format(horoscope_text)
-
-
-    messages = [{"role": "user", "content": prompt}]
-    response = chat(messages)
-    affirmation = response["content"]
+    The structure should be conversational like taylor swift talking to me about my life and pets and not bullet points or third person. Also do not include actions in itallic words, just keep it to conversations.'''.format(horoscope_text,context)
+    
+    # messages = [{"role": "user", "content": prompt}]
+    # response = chat(messages)
+    # affirmation = response["content"]
+    
+    output = llama3_70b(prompt)
+    
     with col1:
          st.markdown(f'<div class="affirmation-box"><p class="horoscope-text">{horoscope_text}</p></div>', unsafe_allow_html=True)
     with col2:
-        st.markdown(f'<div class="affirmation-box"><p class="affirmation-text">{affirmation}</p></div>', unsafe_allow_html=True)
-
-
-groq_api_key = os.getenv['groq_api_token']
-
+        st.markdown(f'<div class="affirmation-box"><p class="affirmation-text">{output}</p></div>', unsafe_allow_html=True)
 
+    
 
+