Update app.py

app.py

@@ -49,45 +49,7 @@ def check_dasha(dasha_type, name, dob, tob, lat, lon, tz):
     if isinstance(response, str):
         return gr.Markdown(response)
     
-    # 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}
-    
-    **Dasha Start Date:** {dasha_start_date}
-    
-    **Remaining Dasha at Birth:** {dasha_remaining_at_birth}
-    
-    ## Mahadasha Sequence:
-    Here is the order of your Mahadasha periods:
-    """
-    
-    if len(mahadasha) == 0 or len(mahadasha_order) == 0:
-        readable_response += "**No Mahadasha data available.**"
-    else:
-        table = "| **Period**   | **Planet**    |\n|--------------|---------------|\n"
-        for i in range(len(mahadasha)):
-            if i < len(mahadasha_order):
-                period = mahadasha_order[i]
-                planet = mahadasha[i]
-                table += f"| {period} | {planet} |\n"
-        
-        readable_response += table
-    
-    if not mahadasha or not mahadasha_order:
-        readable_response += "\n## Astrological Insights: Data not available for the current period."
-    else:
-        readable_response += """
-        ## Astrological Insights:
-        Based on your Mahadasha periods, you can predict the influences of each planet during your life.
-        """
-    
-    return gr.Markdown(readable_response)
+    return gr.Markdown(f"### {dasha_type.replace('-', ' ').title()} Analysis for {name}\n\n" + str(response))
 
 # Gradio UI with improvements
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -122,6 +84,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         create_button("Mahadasha Predictions", "maha-dasha-predictions", check_dasha)
         create_button("Antardasha", "antar-dasha", check_dasha)
         create_button("Char Dasha Current", "char-dasha-current", check_dasha)
+        create_button("Char Dasha Main", "char-dasha-main", check_dasha)
+        create_button("Char Dasha Sub", "char-dasha-sub", check_dasha)
+        create_button("Current Mahadasha Full", "current-mahadasha-full", check_dasha)
+        create_button("Current Mahadasha", "current-mahadasha", check_dasha)
+        create_button("Paryantar Dasha", "paryantar-dasha", check_dasha)
+        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("### 🔍 Enter your details above and click the relevant button to analyze your astrology report!")