Update app.py

app.py

@@ -12,13 +12,13 @@ client = Groq(
     api_key=os.getenv("GROQ_API_KEY"),
 )
 
-st.title("Weather App with Chat Interface")
+# st.title("Weather App with Chat Interface")
 
-input_text = st.text_input("Hi, I am a weather chatbot. Ask me anything!")
+# input_text = st.text_input("Hi, I am a weather chatbot. Ask me anything!")
 
-if st.button("Ask me"):
-    if not input_text:
-        st.error("Please enter a location!")
+# if st.button("Ask me"):
+#     if not input_text:
+#         st.error("Please enter a location!")
 
 # ### Define Dummy Function
 
@@ -42,7 +42,29 @@ def get_current_weather(location):
 
 
 # # define a function as tools
-tools = [
+# tools = [
+#     {
+#         "type": "function",
+#         "function": {
+#             "name": "get_current_weather",
+#             "description": "Get the current weather in a given location",
+#             "parameters": {
+#                 "type": "object",
+#                 "properties": {
+#                     "location": {
+#                         "type": "string",
+#                         "description": "The city and state, e.g. San Francisco, CA"
+#                     }
+#                 },
+#                 "required": ["location"]
+#             }
+#         }
+#     },
+# ]
+
+
+def get_response(input_text):
+    tools = [
     {
         "type": "function",
         "function": {
@@ -61,45 +83,42 @@ tools = [
         }
     },
 ]
-
-
-
-response = client.chat.completions.create(
-    model="mixtral-8x7b-32768",
-    messages=[
-        {
+    response = client.chat.completions.create(
+      model="mixtral-8x7b-32768",
+        messages=[
+          {
             "role": "user",
             "content": input_text,
-        }
-    ],
-    temperature=0,
-    max_tokens=300,
-    tools=tools,
-    tool_choice="auto"
-)
-# print(response)
-# print(response.choices[0].message.content)
+             }
+        ],
+        temperature=0,
+        max_tokens=300,
+        tools=tools,
+        tool_choice="auto"
+    )
+    # print(response)
+    # print(response.choices[0].message.content)
 
-# print(response['choices'][0]['message']['tool_calls'][0]['function']['arguments'])
+    # print(response['choices'][0]['message']['tool_calls'][0]['function']['arguments'])
 
-groq_response = response.choices[0].message
-# print(groq_response)
+    groq_response = response.choices[0].message
+    # print(groq_response)
 
 
-# response.tool_calls[0].function.arguments
+    # response.tool_calls[0].function.arguments
 
-# We can now capture the arguments:
+    # We can now capture the arguments:
 
 
-args = json.loads(groq_response.tool_calls[0].function.arguments)
-# print(args)
+    args = json.loads(groq_response.tool_calls[0].function.arguments)
+    #    print(args)
 
-output=get_current_weather(**args)
-# print(output)
-from groq import Groq
+    output=get_current_weather(**args)
+    # print(output)
+    from groq import Groq
 
-client = Groq()
-completion = client.chat.completions.create(
+    client = Groq()
+    completion = client.chat.completions.create(
     model="mixtral-8x7b-32768",
     messages=[
         {
@@ -116,11 +135,31 @@ completion = client.chat.completions.create(
     top_p=1,
     stream=True,
     stop=None,
-)
-output=""
-st.write("Response:")
-for chunk in completion:
-    output+=chunk.choices[0].delta.content or ""
-
-
-st.write(output)
+    )
+    output=""
+    # st.write("Response:")
+    for chunk in completion:
+        output+=chunk.choices[0].delta.content or ""
+    # output+="\n"
+    return output
+
+def main():
+    st.title("Weather Chatbot")
+
+    # User input
+    st.write("Hi, I am a weather chatbot. Ask me anything!")
+    location = 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)
+            # Display weather details
+            st.json(response)
+        else:
+            st.warning("Please enter a city name.")
+
+if __name__ == "__main__":
+    main()