Update app.py

app.py

@@ -1,8 +1,22 @@
 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 None, None
+
 # 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"
@@ -18,7 +32,11 @@ 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, lat, lon, tz):
+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.")
+    
     response = fetch_astrology_data(f"dosha/{dosha_type}", dob, tob, lat, lon, tz)
     if isinstance(response, str):
         return gr.Markdown(response)
@@ -46,7 +64,11 @@ def check_dosha(dosha_type, name, dob, tob, lat, lon, tz):
     return gr.Markdown(readable_response)
 
 # Function to check Dasha
-def check_dasha(dasha_type, name, dob, tob, lat, lon, tz):
+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.")
+    
     response = fetch_astrology_data(f"dashas/{dasha_type}", dob, tob, lat, lon, tz)
     if isinstance(response, str):
         return gr.Markdown(response)
@@ -70,7 +92,7 @@ def check_dasha(dasha_type, name, dob, tob, lat, lon, tz):
     
     return gr.Markdown(readable_response)
 
-# Gradio UI with improvements
+# Gradio UI with location input for latitude/longitude
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🪐 Astrology Checker - Find Your Dosha & Dasha 🪐")
     
@@ -80,14 +102,13 @@ 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)
+        location = gr.Textbox(label="Location (City or Address)", placeholder="Enter your city or address")
         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)
+        return gr.Button(label, variant="primary").click(fn, inputs=[dosha_type, name, dob, tob, location, tz], outputs=result)
     
     gr.Markdown("## 🌟 Dosha Analysis")
     with gr.Row():
@@ -103,14 +124,6 @@ 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!")