Update app.py

app.py

@@ -1,29 +1,36 @@
 import gradio as gr
+import pytz
+import swisseph as swe
+from datetime import datetime
 from vedicastro.VedicAstro import VedicHoroscopeData
 
-# Function to generate horoscope and predictions
-def generate_horoscope(name, year, month, day, hour, minute, latitude, longitude, utc):
-    vhd = VedicHoroscopeData(
-        year=year, month=month, day=day,
-        hour=hour, minute=minute, second=0,
-        utc=utc, latitude=latitude, longitude=longitude,
-        ayanamsa="Lahiri", house_system="Equal"
-    )
-    
-    chart = vhd.generate_chart()
-    planets_data = vhd.get_planets_data_from_chart(chart)
-    houses_data = vhd.get_houses_data_from_chart(chart)
-    dasa_data = vhd.compute_vimshottari_dasa(chart)
-    
-    prediction = f"### Horoscope for {name}\n\n"
-    prediction += f"#### Planetary Positions\n{planets_data}\n\n"
-    prediction += f"#### House Data\n{houses_data}\n\n"
-    prediction += f"#### Vimshottari Dasa\n{dasa_data}\n\n"
-    
-    return prediction
+# Function to generate horoscope
+def generate_horoscope(name, year, month, day, hour, minute, latitude, longitude, utc_offset):
+    try:
+        vhd = VedicHoroscopeData(
+            year=year, month=month, day=day,
+            hour=hour, minute=minute, second=0,
+            utc=utc_offset, latitude=latitude, longitude=longitude,
+            ayanamsa="Lahiri", house_system="Equal"
+        )
 
-# UI with Gradio
-demo = gr.Interface(
+        chart = vhd.generate_chart()
+        planets_data = vhd.get_planets_data_from_chart(chart)
+        houses_data = vhd.get_houses_data_from_chart(chart)
+        dasa_data = vhd.compute_vimshottari_dasa(chart)
+
+        prediction = f"## Horoscope for {name}\n\n"
+        prediction += f"### 🌟 Planetary Positions\n{planets_data}\n\n"
+        prediction += f"### 🏠 House Data\n{houses_data}\n\n"
+        prediction += f"### 🔮 Vimshottari Dasa (Predictions)\n{dasa_data}\n\n"
+
+        return prediction
+
+    except Exception as e:
+        return f"Error generating horoscope: {str(e)}"
+
+# Define Gradio UI
+iface = gr.Interface(
     fn=generate_horoscope,
     inputs=[
         gr.Textbox(label="Name"),
@@ -32,13 +39,14 @@ demo = gr.Interface(
         gr.Number(label="Day of Birth", value=1),
         gr.Number(label="Hour of Birth", value=12),
         gr.Number(label="Minute of Birth", value=0),
-        gr.Number(label="Latitude", value=28.6139),
+        gr.Number(label="Latitude", value=28.6139),  # Default: New Delhi
         gr.Number(label="Longitude", value=77.2090),
-        gr.Textbox(label="UTC Offset", value="+5:30"),
+        gr.Textbox(label="UTC Offset (e.g., +5.30)", value="+5.30"),
     ],
     outputs=gr.Markdown(),
-    title="AI Vedic Astrology Bot",
-    description="Enter your birth details to generate a detailed Vedic astrology report."
+    title="🔮 AI Vedic Astrology Model",
+    description="Enter your birth details to generate a North Indian-style horoscope chart and AI-driven life predictions."
 )
 
-demo.launch()
+if __name__ == "__main__":
+    iface.launch()