Update app.py

app.py

@@ -70,34 +70,35 @@ def check_dasha(dasha_type, name, dob, tob, lat, lon, tz):
     
     return gr.Markdown(readable_response)
 
-# Function to check Extended Horoscope (Moon Sign, Sun Sign, Ascendant, etc.)
+# Function to check Extended Horoscope
 def check_extended_horoscope(endpoint, name, dob, tob, lat, lon, tz):
     response = fetch_astrology_data(f"extended-horoscope/{endpoint}", dob, tob, lat, lon, tz)
-    if isinstance(response, str):
-        return gr.Markdown(response)
+    if not isinstance(response, dict) or "status" not in response or "response" not in response:
+        return gr.Markdown("Invalid response format.")
+    
+    response_data = response.get("response", {})
+    readable_response = f"""
+    ### {endpoint.replace('-', ' ').title()} Analysis for {name}
+    
+    **Details:**
+    """
+    
+    for key, value in response_data.items():
+        if key == "bot_response":
+            continue
+        readable_response += f"\n- **{key.replace('_', ' ').title()}**: "
+        if isinstance(value, dict):
+            for sub_key, sub_value in value.items():
+                readable_response += f"\n  - {sub_key.replace('_', ' ').title()}: {sub_value}"
+        elif isinstance(value, list):
+            for item in value:
+                readable_response += f"\n  - {item}"
+        else:
+            readable_response += f"{value}"
     
-    readable_response = f"### {endpoint.replace('-', ' ').title()} Analysis for {name}\n\n"
-    
-    if endpoint == "find-moon-sign":
-        readable_response += f"**Moon Sign:** {response.get('moon_sign', 'N/A')}\n**Prediction:** {response.get('prediction', 'No prediction available')}"
-    elif endpoint == "find-sun-sign":
-        readable_response += f"**Sun Sign:** {response.get('sun_sign', 'N/A')}\n**Prediction:** {response.get('prediction', 'No prediction available')}"
-    elif endpoint == "find-ascendant":
-        readable_response += f"**Ascendant:** {response.get('ascendant', 'N/A')}\n**Prediction:** {response.get('prediction', 'No prediction available')}"
-    elif endpoint == "current-sade-sati":
-        readable_response += f"**Sade Sati Period:** {response.get('shani_period_type', 'N/A')}\n**Description:** {response.get('description', 'No description available')}"
-    elif endpoint == "extended-kundli-details":
-        readable_response += f"**Extended Kundli Details:** {response.get('kundli_details', 'N/A')}"
-    elif endpoint == "shad-bala":
-        readable_response += f"**Shad Bala Score:** {response.get('shad_bala', 'N/A')}\n**Prediction:** {response.get('prediction', 'No prediction available')}"
-    
-    if "remedies" in response:
-        readable_response += "\n\n**Remedies:**"
-        for remedy in response['remedies']:
-            readable_response += f"\n- {remedy}"
-
     return gr.Markdown(readable_response)
 
+
 # Gradio UI with new buttons for extended horoscope
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🪐 Astrology Checker - Find Your Dosha, Dasha, Moon Sign, Sun Sign, Ascendant, Sade Sati, Kundli & Shad Bala 🪐")