Update app.py

app.py

@@ -1,22 +1,8 @@
 import gradio as gr
 import requests
-from geopy.geocoders import Nominatim
 
 API_KEY = "d6cf6f2e-301e-5f63-a988-b0a3aefd0896"
 
-# Initialize Geolocator
-geolocator = Nominatim(user_agent="astrology_checker")
-
-# Function to fetch latitude and longitude based on location
-def get_lat_lon(location):
-    location_info = geolocator.geocode(location)
-    if location_info:
-        lat = location_info.latitude
-        lon = location_info.longitude
-        return lat, lon
-    else:
-        return 0, 0
-
 # Function to fetch astrology data
 def fetch_astrology_data(endpoint, dob, tob, lat, lon, tz):
     url = f"https://api.vedicastroapi.com/v3-json/{endpoint}?dob={dob}&tob={tob}&lat={lat}&lon={lon}&tz={tz}&api_key={API_KEY}&lang=en"
@@ -32,11 +18,7 @@ def fetch_astrology_data(endpoint, dob, tob, lat, lon, tz):
         return f"Error fetching data: {response.status_code} - {response.text}"
 
 # Function to check Dosha
-def check_dosha(dosha_type, name, dob, tob, location, tz):
-    lat, lon = get_lat_lon(location)
-    if lat is None or lon is None:
-        return gr.Markdown("Location not found. Please check the location and try again.")
-    
+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 gr.Markdown(response)
@@ -64,11 +46,7 @@ def check_dosha(dosha_type, name, dob, tob, location, tz):
     return gr.Markdown(readable_response)
 
 # Function to check Dasha
-def check_dasha(dasha_type, name, dob, tob, location, tz):
-    lat, lon = get_lat_lon(location)
-    if lat is None or lon is None:
-        return gr.Markdown("Location not found. Please check the location and try again.")
-    
+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 gr.Markdown(response)
@@ -92,7 +70,7 @@ def check_dasha(dasha_type, name, dob, tob, location, tz):
     
     return gr.Markdown(readable_response)
 
-# Gradio UI with location input for latitude/longitude
+# Gradio UI with improvements
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🪐 Astrology Checker - Find Your Dosha & Dasha 🪐")
     
@@ -102,15 +80,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         tob = gr.Textbox(label="Time of Birth (HH:MM)", placeholder="11:40")
     
     with gr.Row():
-        location = gr.Textbox(label="Location (City or Address)", placeholder="Enter your city or address")
+        lat = gr.Number(label="Latitude", value=11)
+        lon = gr.Number(label="Longitude", value=77)
         tz = gr.Number(label="Timezone (e.g., 5.5 for India)", value=5.5)
     
     result = gr.Markdown()
-
-    # Function to create a button with dosha_type as gr.State
+    
     def create_button(label, dosha_type, fn):
-        return gr.Button(label, variant="primary").click(fn, inputs=[name, dob, tob, location, tz, gr.State(dosha_type)], outputs=result)
-
+        return gr.Button(label, variant="primary").click(fn, inputs=[gr.State(dosha_type), name, dob, tob, lat, lon, tz], outputs=result)
+    
     gr.Markdown("## 🌟 Dosha Analysis")
     with gr.Row():
         create_button("Mangal Dosh", "mangal-dosh", check_dosha)
@@ -125,6 +103,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!")