Update app.py

app.py

@@ -70,9 +70,37 @@ def check_dasha(dasha_type, name, dob, tob, lat, lon, tz):
     
     return gr.Markdown(readable_response)
 
-# Gradio UI with improvements
+# Function to check Extended Horoscope (Moon Sign, Sun Sign, Ascendant, etc.)
+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)
+    
+    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 🪐")
+    gr.Markdown("# 🪐 Astrology Checker - Find Your Dosha, Dasha, Moon Sign, Sun Sign, Ascendant, Sade Sati, Kundli & Shad Bala 🪐")
     
     with gr.Row():
         name = gr.Textbox(label="Name", placeholder="Enter your name")
@@ -86,8 +114,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     result = gr.Markdown()
     
-    def create_button(label, dosha_type, fn):
-        return gr.Button(label, variant="primary").click(fn, inputs=[gr.State(dosha_type), name, dob, tob, lat, lon, tz], outputs=result)
+    def create_button(label, endpoint, fn):
+        return gr.Button(label, variant="primary").click(fn, inputs=[name, dob, tob, lat, lon, tz], outputs=result)
     
     gr.Markdown("## 🌟 Dosha Analysis")
     with gr.Row():
@@ -111,6 +139,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         create_button("Specific Dasha", "specific-sub-dasha", check_dasha)
         create_button("Yogini Dasha Main", "yogini-dasha-main", check_dasha)
         create_button("Yogini Dasha Sub", "yogini-dasha-sub", check_dasha)
+
+    gr.Markdown("## 🌙 Extended Horoscope")
+    with gr.Row():
+        create_button("Find Moon Sign", "find-moon-sign", check_extended_horoscope)
+        create_button("Find Sun Sign", "find-sun-sign", check_extended_horoscope)
+        create_button("Find Ascendant", "find-ascendant", check_extended_horoscope)
+        create_button("Current Sade Sati", "current-sade-sati", check_extended_horoscope)
+        create_button("Extended Kundli Details", "extended-kundli-details", check_extended_horoscope)
+        create_button("Shad Bala", "shad-bala", check_extended_horoscope)
     
     gr.Markdown("### 🔍 Enter your details above and click the relevant button to analyze your astrology report!")