Update app.py

app.py

@@ -44,6 +44,26 @@ def load_csv_data():
 
 # Load CSV data
 csv_data = load_csv_data()
+WEATHER_API_KEY = "ee75ffd59875aa5ca6c207e594336b30"
+
+def get_weather(city: str):
+    """Fetches weather data from OpenWeatherMap API."""
+    url = f"https://api.openweathermap.org/data/2.5/weather?q={city}&appid={WEATHER_API_KEY}&units=metric"
+    try:
+        response = requests.get(url, timeout=5)
+        response.raise_for_status()
+        data = response.json()
+
+        if data.get("cod") == 200:
+            weather_description = data['weather'][0]['description']
+            temperature = data['main']['temp']
+            humidity = data['main']['humidity']
+            pressure = data['main']['pressure']
+            return temperature, humidity, weather_description, pressure
+    except Exception as e:
+        print(f"Error fetching weather data: {e}")
+
+    return None, None, None, None
 
 class RetellRequest(BaseModel):
     name: str  # Function name
@@ -73,55 +93,40 @@ def search_csv_data(df: pd.DataFrame, search_terms: Dict[str, str]) -> pd.DataFr
     return result
 
 @app.post("/api/market-prices")
-async def market_prices_endpoint(
-    request: Request,
-    x_retell_signature: str = Header(None, alias="X-Retell-Signature")
-):
-    """Handle market prices function call from Retell.ai"""
-    request_body = await request.body()
-    retell_request = json.loads(request_body.decode('utf-8'))
-    
-    # Extract arguments
-    crop_name = retell_request["args"].get("crop_name", "")
-    state = retell_request["args"].get("state", "")
-    district = retell_request["args"].get("district", "")
-    
-    try:
-        # Check if we have relevant data in CSVs
-        crop_data = None
-        if not csv_data['crop_advisory'].empty:
-            search_terms = {'crop_name': crop_name}
-            if state:
-                search_terms['state'] = state
-            crop_data = search_csv_data(csv_data['crop_advisory'], search_terms)
-        
-        # Mock response - replace with actual API integration
-        if not crop_data.empty:
-            # Use data from CSV if available
-            base_price = "₹2,180 per quintal"
-            location_text = f" in {district or ''} {state}".strip()
-        else:
-            base_price = "₹2,180 per quintal"
-            location_text = f" in {district or ''} {state}".strip()
-        
-        response_data = {
-            "current_price": f"{base_price} for {crop_name}",
-            "trend": "Prices increased by ₹50 in last week",
-            "msp": "₹2,125 per quintal",
-            "advice": "Current prices are above MSP. Good time to sell.",
-            "location": location_text
-        }
-        
-        return {
-            "result": f"The current price of {crop_name}{location_text} is {response_data['current_price']}. {response_data['trend']}. MSP is {response_data['msp']}. {response_data['advice']}",
-            "response_data": response_data
-        }
-        
-    except Exception as e:
-        return {
-            "result": f"I'm sorry, I couldn't get the current market prices for {crop_name}. Please try again later or contact your local mandi for current rates.",
-            "error": str(e)
-        }
+async def market_prices(request: dict):
+    crop_name = request.get("query", {}).get("crop_name", "").strip()
+    state = request.get("query", {}).get("state", "").strip()
+    district = request.get("query", {}).get("district", "").strip()
+
+    if "crop_advisory" in csv_data:
+        df = csv_data["crop_advisory"]
+        mask = (
+            df["crop_name"].str.lower() == crop_name.lower()
+        ) & (df["state"].str.lower() == state.lower()) & (
+            df["district"].str.lower() == district.lower()
+        )
+        matches = df[mask]
+
+        if not matches.empty:
+            try:
+                avg_price = matches["price"].astype(float).mean()
+                result = (
+                    f"The average market price of {crop_name} in {district}, {state} "
+                    f"is ₹{avg_price:.2f} per quintal."
+                )
+            except Exception:
+                result = f"Market data found but price column not numeric for {crop_name} in {district}, {state}."
+
+            return {
+                "success": True,
+                "result": result,
+                "data": matches.to_dict(orient="records")
+            }
+
+    return {
+        "success": False,
+        "result": f"No market price data found for {crop_name} in {district}, {state}."
+    }
 
 @app.post("/api/scheme-eligibility") 
 async def scheme_eligibility_endpoint(
@@ -207,42 +212,43 @@ async def scheme_eligibility_endpoint(
         }
 
 @app.post("/api/weather-advisory")
-async def weather_advisory_endpoint(
-    request: Request,
-    x_retell_signature: str = Header(None, alias="X-Retell-Signature")
-):
-    """Handle weather advisory function call from Retell.ai"""
-    request_body = await request.body()
-    retell_request = json.loads(request_body.decode('utf-8'))
-    
-    location = retell_request["args"].get("location", "")
-    state = retell_request["args"].get("state", "")
-    advisory_type = retell_request["args"].get("advisory_type", "general")
-    
-    try:
-        # Mock weather data - integrate with IMD API
-        weather_info = {
-            "forecast": "Light to moderate rainfall expected in next 2-3 days",
-            "temperature": "Maximum 28°C, Minimum 18°C",
-            "advisory": "Avoid irrigation for next 3 days due to expected rainfall. Good time for land preparation for rabi sowing.",
-            "warning": None
-        }
-        
-        advisory_text = f"Weather forecast for {location}, {state}: {weather_info['forecast']}. Temperature: {weather_info['temperature']}. Advisory: {weather_info['advisory']}"
-        
-        if weather_info["warning"]:
-            advisory_text += f" Warning: {weather_info['warning']}"
-        
-        return {
-            "result": advisory_text,
-            "weather_data": weather_info
-        }
-        
-    except Exception as e:
-        return {
-            "result": f"I couldn't get weather information for {location} right now. Please check local weather updates or contact your agriculture department.",
-            "error": str(e)
+async def weather_advisory(request: dict):
+    city = request.get("query", {}).get("location", "").strip()
+
+    temperature, humidity, description, pressure = get_weather(city)
+    if temperature is None:
+        # Fallback values
+        temperature, humidity, description, pressure = 32.0, 60, "Not Available", 1012
+        weather_condition = "NORMAL"
+    else:
+        desc_lower = description.lower()
+        if "clear" in desc_lower:
+            weather_condition = "SUNNY"
+        elif "rain" in desc_lower:
+            weather_condition = "RAINY"
+        elif "wind" in desc_lower:
+            weather_condition = "WINDY"
+        else:
+            weather_condition = "NORMAL"
+
+    result = (
+        f"Weather in {city}: {description}. "
+        f"Temperature {temperature}°C, Humidity {humidity}%, Pressure {pressure} hPa. "
+        f"Condition classified as {weather_condition}."
+    )
+
+    return {
+        "success": True,
+        "result": result,
+        "data": {
+            "city": city,
+            "temperature": temperature,
+            "humidity": humidity,
+            "pressure": pressure,
+            "description": description,
+            "condition": weather_condition
         }
+    }
 
 @app.post("/api/crop-advisory")
 async def crop_advisory_endpoint(