Update app.py

app.py

@@ -8,198 +8,137 @@ import requests
 from bs4 import BeautifulSoup
 
 
-# def ollama():
-#     os.environ['OLLAMA_HOST'] = '0.0.0.0:11434'
-#     os.environ['OLLAMA_ORIGINS'] = '*'
-#     print("Starting Ollama server...")
-#     try:
-#         subprocess.Popen(["ollama", "serve"])
-#         print("Ollama server started.")
-#     except Exception as e:
-#         print(f"Failed to start Ollama server: {e}")
-
-# ollama_thread = threading.Thread(target=ollama)
-# ollama_thread.start()
-
-# model = 'llama3'
-
-# def chat(messages):
-#     r = requests.post(
-#         "http://0.0.0.0:11434/api/chat",
-#         json={"model": model, "messages": messages, "stream": True},
-#     )
-#     r.raise_for_status()
-#     output = ""
-#     for line in r.iter_lines():
-#         body = json.loads(line)
-#         if "error" in body:
-#             raise Exception(body["error"])
-#         if body.get("done") is False:
-#             message = body.get("message", "")
-#             content = message.get("content", "")
-#             output += content
-#             print(content, end="", flush=True)
-#         if body.get("done", False):
-#             message["content"] = output
-#             return message
+
 
 st.set_page_config(page_title="Affirmations", page_icon=":muscle:", layout="wide")
 
-# st.markdown(
-#     """
-#     <style>
-#     .stApp {
-#         background-color: #e9e5d9;
-#     }
-#     .title {
-#         font-size: 36px;
-#         font-weight: bold;
-#         text-align: center;
-#         margin-bottom: 20px;
-#     }
-#     .image-container {
-#         display: flex;
-#         justify-content: center;
-#         margin-bottom: 20px;
-#     }
-#     .affirmation-box {
-#         background-color: #ffffff;
-#         padding: 20px;
-#         border-radius: 10px;
-#         margin-top: 20px;
-#         text-align: center;
-#         box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
-#     }
-#     .affirmation-text {
-#         font-family: 'Helvetica', sans-serif;
-#         font-size: 18px;
-#         color: #333333;
-#     }
-#     .stButton button {
-#         background-color: #ff0000;
-#         color: #ffffff;
-#         font-size: 16px;
-#         padding: 10px 20px;
-#         border-radius: 5px;
-#         margin-top: 20px;
-#         width: 100%;
-#     }
-#     </style>
-#     """,
-#     unsafe_allow_html=True
-# )
-
-# col1, col2 = st.columns(2)
-
-# st.title("Taylor Swift's Gym Affirmations")
-
-# with col2:
-#     st.image("https://www.rollingstone.com/wp-content/uploads/2023/05/GettyImages-1482443875.jpeg?w=1581&h=1054&crop=1")
-
-# 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.
-#     # """
-
-
-
-#     def get_astrology_sign(birth_date):
-#         # Zodiac signs divided by the date ranges
-#         zodiac_dates = [
-#             ((1, 20), (2, 18), "11"),
-#             ((2, 19), (3, 20), "12"),
-#             ((3, 21), (4, 19), "1"),
-#             ((4, 20), (5, 20), "2"),
-#             ((5, 21), (6, 20), "3"),
-#             ((6, 21), (7, 22), "4"),
-#             ((7, 23), (8, 22), "5"),
-#             ((8, 23), (9, 22), "6"),
-#             ((9, 23), (10, 22), "7"),
-#             ((10, 23), (11, 21), "8"),
-#             ((11, 22), (12, 21), "9"),
-#             ((12, 22), (1, 19), "10")
-#         ]
+st.markdown(
+    """
+    <style>
+    .stApp {
+        background-color: #e9e5d9;
+    }
+    .title {
+        font-size: 36px;
+        font-weight: bold;
+        text-align: center;
+        margin-bottom: 20px;
+    }
+    .image-container {
+        display: flex;
+        justify-content: center;
+        margin-bottom: 20px;
+    }
+    .affirmation-box {
+        background-color: #ffffff;
+        padding: 20px;
+        border-radius: 10px;
+        margin-top: 20px;
+        text-align: center;
+        box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+    }
+    .affirmation-text {
+        font-family: 'Helvetica', sans-serif;
+        font-size: 18px;
+        color: #333333;
+    }
+    .stButton button {
+        background-color: #ff0000;
+        color: #ffffff;
+        font-size: 16px;
+        padding: 10px 20px;
+        border-radius: 5px;
+        margin-top: 20px;
+        width: 100%;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
+col1, col2 = st.columns(2)
+
+st.title("Taylor Swift's Gym Affirmations")
+
+with col2:
+    st.image("https://www.rollingstone.com/wp-content/uploads/2023/05/GettyImages-1482443875.jpeg?w=1581&h=1054&crop=1")
+
+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.
+    # """
+
+
+
+    def get_astrology_sign(birth_date):
+        # Zodiac signs divided by the date ranges
+        zodiac_dates = [
+            ((1, 20), (2, 18), "11"),
+            ((2, 19), (3, 20), "12"),
+            ((3, 21), (4, 19), "1"),
+            ((4, 20), (5, 20), "2"),
+            ((5, 21), (6, 20), "3"),
+            ((6, 21), (7, 22), "4"),
+            ((7, 23), (8, 22), "5"),
+            ((8, 23), (9, 22), "6"),
+            ((9, 23), (10, 22), "7"),
+            ((10, 23), (11, 21), "8"),
+            ((11, 22), (12, 21), "9"),
+            ((12, 22), (1, 19), "10")
+        ]
         
-#         month = birth_date.month
-#         day = birth_date.day
+        month = birth_date.month
+        day = birth_date.day
 
-#         for start, end, sign in zodiac_dates:
-#             if (month == start[0] and day >= start[1]) or (month == end[0] and day <= end[1]):
-#                 return sign
-#         return "Unknown"   
+        for start, end, sign in zodiac_dates:
+            if (month == start[0] and day >= start[1]) or (month == end[0] and day <= end[1]):
+                return sign
+        return "Unknown"   
 
     
-#     def get_horoscope(zodiac_sign):
-#             day = 'tomorrow'
-#             url = (
-#                 "https://www.horoscope.com/us/horoscopes/general/"
-#                 f"horoscope-general-daily-{day}.aspx?sign={zodiac_sign}"
-#             )
-#             soup = BeautifulSoup(requests.get(url).content, "html.parser")
-#             horoscope_text = soup.find("div", class_="main-horoscope").p.text
-#             return horoscope_text
-
-# # Cancer
+    def get_horoscope(zodiac_sign):
+            day = 'tomorrow'
+            url = (
+                "https://www.horoscope.com/us/horoscopes/general/"
+                f"horoscope-general-daily-{day}.aspx?sign={zodiac_sign}"
+            )
+            soup = BeautifulSoup(requests.get(url).content, "html.parser")
+            horoscope_text = soup.find("div", class_="main-horoscope").p.text
+            return horoscope_text
+
+# Cancer
     
-#     owner_sign = "4"
-#     horoscope_text = get_horoscope(owner_sign)
+    owner_sign = "4"
+    horoscope_text = get_horoscope(owner_sign)
     
-#     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\n
-#     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
-#     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"]
-#     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)
-
-
-# from langchain_community.llms import Ollama
-# llm = Ollama(model="llama3")
-
-# 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\n
-# 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
-# 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
-# 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.'''
-
-# import os
-# response = llm.invoke(prompt)
-
-# st.markdown(response)
-
-from langchain_community.llms import Ollama
-
-# Initialize the Ollama model
-llm = Ollama(model="llama3")
-
-# Define your prompt
-prompt = '''tell me a joke.'''
-
-# Invoke the model with your prompt
-response = llm.invoke(prompt)
-
-# Display the response in your Streamlit app
-st.markdown(response)
+    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\n
+    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
+    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"]
+    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']