Update app.py

app.py

@@ -1,24 +1,58 @@
-import gradio as gr
 import os
+import json
+import gradio as gr
 import requests
-from groq import Groq 
+from dotenv import load_dotenv
+from groq import Groq  # Correct import after installing groq module
 
-def preprocess_city_name_and_generate_insight(city_name):
-    api_key = os.getenv("GROQ_API_KEY")
-    if not api_key:
-        return city_name, "GROQ API key is not set in the environment variables."
+# Load environment variables
+load_dotenv()
 
-    try:
-        client = Groq(api_key=api_key)
+# Groq client initialization
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-        prompt = f"Generate a short, interesting fact about {city_name} in less than 10 words."
-        result = client.query(prompt=prompt, model_id="llama3-70b-8192")  # Adjust parameters as necessary
+def preprocess_city_name_and_generate_insight(city_name):
+    tools = [
+        {
+            "type": "function",
+            "function": {
+                "name": "generate_city_insight",
+                "description": "Generate a short, interesting fact about a specified city",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "city_name": {
+                            "type": "string",
+                            "description": "The name of the city to generate insights for",
+                        },
+                    },
+                    "required": ["city_name"],
+                },
+            },   
+        }
+    ]
 
-        insight = result.get('generated_text', "No insight generated.").strip()
-    except Exception as e:
-        print(f"Error generating insight: {str(e)}")
-        insight = "Could not generate insight due to an error."
+    # Simulate a user message asking for insights about a city
+    response = client.chat.completions.create(
+        model="mixtral-8x7b-32768",
+        messages=[
+            {
+                "role": "user",
+                "content": f"What interesting fact can you tell me about {city_name}?",
+            }
+        ],
+        temperature=0,
+        max_tokens=300,
+        tools=tools,
+        tool_choice="auto"
+    )
 
+    # Extract and handle the response
+    groq_response = response.choices[0].message
+    args = json.loads(groq_response.tool_calls[0].function.arguments)
+
+    # Assuming generate_city_insight function is defined somewhere that returns an insight
+    insight = generate_city_insight(**args)
     return city_name, insight
 
 def get_weather(city_name):
@@ -26,10 +60,10 @@ def get_weather(city_name):
     API_Key = os.getenv("OPENWEATHER_API_KEY")
     if not API_Key:
         return "API Key is not set. Please set the OPENWEATHER_API_KEY environment variable."
-    
+
     url = f'https://api.openweathermap.org/data/2.5/weather?q={corrected_city_name}&appid={API_Key}&units=metric'
     response = requests.get(url)
-    
+
     if response.status_code == 200:
         data = response.json()
         weather = data['weather'][0]['description']
@@ -43,7 +77,7 @@ def get_weather(city_name):
     else:
         return "Failed to retrieve data. Please check the city name and try again."
 
-# Defining Gradio interface
+# Gradio interface definition
 iface = gr.Interface(
     fn=get_weather,
     inputs=gr.Textbox(label="Enter City Name", placeholder="Type here..."),