Update app.py

app.py

@@ -1,4 +1,3 @@
-
 # # Function Calling with OpenAI APIs
 import requests
 import os
@@ -48,13 +47,13 @@ def get_current_weather(location):
 #         "type": "function",
 #         "function": {
 #             "name": "get_current_weather",
-#             "description": "Get the current weather details for a given location.",
+#             "description": "Get the current weather in a given location",
 #             "parameters": {
 #                 "type": "object",
 #                 "properties": {
 #                     "location": {
 #                         "type": "string",
-#                         "description": "The city and country code in ISO 3166 format for example, London, UK or Bengaluru, IN"
+#                         "description": "The city and state, e.g. San Francisco, CA"
 #                     }
 #                 },
 #                 "required": ["location"]
@@ -71,13 +70,13 @@ def get_response(input_text,client):
         "type": "function",
         "function": {
             "name": "get_current_weather",
-            "description": "Get the current weather in a given location",
+            "description": "Get the real time weather of a given location",
             "parameters": {
                 "type": "object",
                 "properties": {
                     "location": {
                         "type": "string",
-                        "description": "The city and state, e.g. San Francisco, CA"
+                        "description": "The city, state and country, e.g. San Francisco, CA, US"
                     }
                 },
                 "required": ["location"]
@@ -89,8 +88,12 @@ def get_response(input_text,client):
       model="mixtral-8x7b-32768",
         messages=[
           {
+            "role": "system",
+            "content": "You are a helpful assistant. You will asked details regarding the weather. If there is no question with respect to a weather, show a polite error message."
+        },
+        {
             "role": "user",
-            "content": input_text,
+            "content": input_text+" Use a tool given to you",
              }
         ],
         temperature=0,
@@ -108,6 +111,10 @@ def get_response(input_text,client):
         return "No response from Groq"
     else:
         groq_response = response.choices[0].message
+        args = json.loads(groq_response.tool_calls[0].function.arguments)
+    #    print(args)
+
+        output=get_current_weather(**args)
     # print(groq_response)
 
 
@@ -116,10 +123,7 @@ def get_response(input_text,client):
     # We can now capture the arguments:
 
 
-    args = json.loads(groq_response.tool_calls[0].function.arguments)
-    #    print(args)
 
-    output=get_current_weather(**args)
     # print(output)
     from groq import Groq
 
@@ -129,7 +133,7 @@ def get_response(input_text,client):
     messages=[
         {
             "role": "system",
-            "content": "You are a helpful assistant. You are given the weather details in json format. Read the data and give a brief description of the weather and then answer the question. All temperatures are in kelvin. Only mention details about the weather. "
+            "content": "You are a helpful assistant. You are given the weather details in json format. Read the data and give a brief description of the weather and then answer the question. All temperatures are in kelvin. Only mention details about the weather. If the JSON mentions data fetch error, only give a polite error message and tell the user to ask questions related to the weather of a place. "
         },
         {
             "role": "user",
@@ -150,26 +154,29 @@ def get_response(input_text,client):
     return output
 
 def main():
-    st.title("Weather Wizard")
+    st.title("WeatherWizard")
+    st.caption("Ask about any location's weather in plain English, get answers faster than you can say umbrella")
     client = Groq(
         api_key=os.getenv("GROQ_API_KEY"),
         )
     # User input
-    st.write("Hi, I am a weather chatbot. Ask me anything!")
-    location = st.text_input("Type in your question")
+    # st.write("Hi, I am a weather chatbot. Ask me anything!")
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]
+
+    for msg in st.session_state.messages:
+        st.chat_message(msg["role"]).write(msg["content"])
+        
+    # input_text = st.text_input("Type in your question")
 
     # Ask me button
-    if st.button("Ask me"):
-        # Check if location is provided
-        if location:
-            # Get current weather
-            response = get_response(location,client)
-            # Display weather details
-            # st.json(response)
-            st.write(response)
-        else:
-            st.warning("Please enter a city name.")
+    if prompt := st.chat_input():
+    
 
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        st.chat_message("user").write(prompt)
+        response = get_response(prompt,client)
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        st.chat_message("assistant").write(response)
 if __name__ == "__main__":
-    main()
-
+    main()