Update app.py

app.py

@@ -17,54 +17,39 @@ def fetch_astrology_data(endpoint, dob, tob, lat, lon, tz):
     else:
         return f"Error fetching data: {response.status_code}"
 
-# Function to check dosha
-def check_dosha(dosha_type, name, dob, tob, lat, lon, tz):
-    response = fetch_astrology_data(f"dosha/{dosha_type}", dob, tob, lat, lon, tz)
-    if isinstance(response, str):
-        return response
-    
-    factors = response.get("factors", {})
-    dosha_present = response.get("is_dosha_present", False)
-    bot_response = response.get("bot_response", "No information available")
-    score = response.get("score", "N/A")
-    
-    readable_response = f"""
-    ### {dosha_type.replace('-', ' ').title()} Analysis for {name}
-    
-    **Dosha Presence:** {'Yes' if dosha_present else 'No'}
-    **Score:** {score}%
-    
-    **Detailed Analysis:**
-    - **Moon Influence:** {factors.get('moon', 'Not mentioned')}
-    - **Saturn Influence:** {factors.get('saturn', 'Not mentioned')}
-    - **Rahu Influence:** {factors.get('rahu', 'Not mentioned')}
-    
-    **Astrological Suggestion:** {bot_response}
-    """
-    return readable_response
-
 # Function to check dasha
 def check_dasha(dasha_type, name, dob, tob, lat, lon, tz):
     response = fetch_astrology_data(f"dashas/{dasha_type}", dob, tob, lat, lon, tz)
     if isinstance(response, str):
         return response
     
-    # Format the dasha data to make it more readable
-    analysis = response.get("analysis", [])
-    period = response.get("period", "N/A")
-    dasha_details = "\n".join([f"- {item}" for item in analysis])
+    # Extracting Mahadasha and its order details
+    mahadasha = response.get("mahadasha", [])
+    mahadasha_order = response.get("mahadasha_order", [])
+    dasha_start_date = response.get("dasha_start_date", "N/A")
+    dasha_remaining_at_birth = response.get("dasha_remaining_at_birth", "N/A")
     
+    # Format the dasha data to make it more readable
     readable_response = f"""
     ### {dasha_type.replace('-', ' ').title()} Analysis for {name}
     
-    **Period:** {period}
+    **Dasha Start Date:** {dasha_start_date}
     
-    **Detailed Dasha Analysis:**
-    {dasha_details}
+    **Remaining Dasha at Birth:** {dasha_remaining_at_birth}
     
-    **Astrological Suggestions:** 
-    - Follow the guidance based on your dasha period.
+    ## Mahadasha Sequence:
+    Here is the order of your Mahadasha periods:
+    
+    | **Period**   | **Planet**    |
+    |--------------|---------------|
     """
+    
+    for i in range(len(mahadasha)):
+        if i < len(mahadasha_order):
+            readable_response += f"| {mahadasha_order[i]} | {mahadasha[i]} |\n"
+    
+    readable_response += "\n**Astrological Insights:** Based on your Mahadasha periods, you can predict the influences of each planet during your life."
+    
     return readable_response
 
 # Gradio UI with improvements
@@ -77,22 +62,22 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         tob = gr.Textbox(label="Time of Birth (HH:MM)", placeholder="11:40")
     
     with gr.Row():
-        lat = gr.Number(label="Latitude", value=11)
-        lon = gr.Number(label="Longitude", value=77)
+        lat = gr.Number(label="Latitude", value=0)
+        lon = gr.Number(label="Longitude", value=0)
         tz = gr.Number(label="Timezone (e.g., 5.5 for India)", value=5.5)
     
     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, dasha_type, fn):
+        return gr.Button(label, variant="primary").click(fn, inputs=[gr.State(dasha_type), name, dob, tob, lat, lon, tz], outputs=result)
     
     gr.Markdown("## 🌟 Dosha Analysis")
     with gr.Row():
-        create_button("Mangal Dosh", "mangal-dosh", check_dosha)
-        create_button("Kaalsarp Dosh", "kaalsarp-dosh", check_dosha)
-        create_button("Manglik Dosh", "manglik-dosh", check_dosha)
-        create_button("Pitra Dosh", "pitra-dosh", check_dosha)
-        create_button("Papasamaya", "papasamaya", check_dosha)
+        create_button("Mangal Dosh", "mangal-dosh", check_dasha)
+        create_button("Kaalsarp Dosh", "kaalsarp-dosh", check_dasha)
+        create_button("Manglik Dosh", "manglik-dosh", check_dasha)
+        create_button("Pitra Dosh", "pitra-dosh", check_dasha)
+        create_button("Papasamaya", "papasamaya", check_dasha)
     
     gr.Markdown("## 🔮 Dasha Analysis")
     with gr.Row():