Update app.py

app.py

@@ -44,32 +44,40 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def get_horoscope(sign: str, date: str = None, language: str = None) -> str:
+def get_horoscope(sign: str, date: str = None, language: str = "EN") -> str:
     """Fetches the horoscope for a given zodiac sign and date.
-    If the user does not provide a date, use today's real date.
+    Uses the ExaWeb API. Defaults to today if no date is provided.
+    
     Args:
         sign: Zodiac sign (e.g., Aries, Taurus, Gemini)
         date: Date in any format (optional)
+        language: Language code ('EN' or 'HI')
     """
-    date = parse_date(date)
-    # Placeholder logic
-    if not date:
-        date = datetime.now().strftime("%Y-%m-%d")
-
-    url = "https://api.exaweb.in:3004/api/rashi"
     try:
-        response = requests.get(url, params={"language": language, "rashi": sign.upper(), "date": date})
+        # Parse or use today's date
+        if date:
+            date_obj = parse_date(date)
+        else:
+            date_obj = datetime.now()
+        formatted_date = date_obj.strftime("%d-%m-%Y")
+
+        # API call
+        url = "https://api.exaweb.in:3004/api/rashi/getForAllLangs"
+        response = requests.get(url, params={"day": formatted_date, "language": language})
         response.raise_for_status()
         data = response.json()
 
-        # Extract the horoscope
-        horoscope = data.get("rashiData", {}).get("prediction") or "No prediction found."
-        today = datetime.now().strftime("%Y-%m-%d")
+        # Find horoscope for the given sign
+        sign_upper = sign.upper()
+        for item in data.get("data", []):
+            if item.get("sign", "").upper() == sign_upper:
+                prediction = item.get("prediction", "No prediction found.")
+                return f"Horoscope for {sign.capitalize()} on {formatted_date}:\n{prediction}"
 
-        return f"Horoscope for {sign.capitalize()} on {today}:\n{horoscope}"
+        return f"No horoscope found for sign: {sign}"
 
     except Exception as e:
-        return f"Failed to fetch horoscope for {sign}: {str(e)}"
+        return f"Error fetching horoscope: {str(e)}"
 
 @tool
 def get_date_panchang(date: str = None) -> str: