Update app.py

app.py

@@ -63,17 +63,20 @@ with gr.Blocks() as demo:
     
     result = gr.Markdown()
     
+    def create_button(label, dosha_type, fn):
+        return gr.Button(label).click(fn, inputs=[gr.State(dosha_type), name, dob, tob, lat, lon, tz], outputs=result)
+    
     with gr.Row():
-        gr.Button("Mangal Dosh").click(check_dosha, inputs=["mangal-dosh", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Kaalsarp Dosh").click(check_dosha, inputs=["kaalsarp-dosh", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Manglik Dosh").click(check_dosha, inputs=["manglik-dosh", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Pitra Dosh").click(check_dosha, inputs=["pitra-dosh", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Papasamaya").click(check_dosha, inputs=["papasamaya", name, dob, tob, lat, lon, tz], outputs=result)
+        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)
     
     with gr.Row():
-        gr.Button("Mahadasha").click(check_dasha, inputs=["maha-dasha", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Mahadasha Predictions").click(check_dasha, inputs=["maha-dasha-predictions", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Antardasha").click(check_dasha, inputs=["antar-dasha", name, dob, tob, lat, lon, tz], outputs=result)
-        gr.Button("Char Dasha Current").click(check_dasha, inputs=["char-dasha-current", name, dob, tob, lat, lon, tz], outputs=result)
+        create_button("Mahadasha", "maha-dasha", check_dasha)
+        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)
     
     demo.launch()