Update app.py

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 import openai
 import os
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
 
 # Configure OpenAI API for Groq
 openai.api_key = os.getenv("API_KEY")
@@ -8,70 +12,122 @@ openai.api_base = "https://api.groq.com/openai/v1"
 
 def get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac_sign, query):
     try:
+        if not openai.api_key:
+            return "🔴 Error: API key not found. Please check your configuration."
+            
+        system_prompt = f"""You are Master Celestia, an AI astrologer with 500 years of mystical wisdom. Analyze the birth chart for:
+        Name: {name}
+        DOB: {dob}
+        Time: {time_of_birth}
+        Place: {place_of_birth}
+        Zodiac: {zodiac_sign}
+
+        Provide a professional analysis including:
+        1. Current planetary transits and aspects
+        2. Career and financial outlook (next 3 months)
+        3. Relationship compatibility insights
+        4. Personalized recommendations (crystals, mantras, colors)
+        5. Weekly guidance based on moon phases
+
+        Format with emojis and section headers. Keep responses under 600 tokens."""
+
         messages = [
-            {"role": "system", "content": f"""
-                You are a professional astrologer with mystical wisdom. Analyze the birth chart for {name} born on {dob} 
-                at {time_of_birth} in {place_of_birth}. Consider their {zodiac_sign} sun sign and current planetary 
-                transits. Provide detailed, personalized insights in a mystical yet clear tone. Include:
-                - Key planetary aspects
-                - Strengths and challenges
-                - Career and relationship guidance
-                - Personalized recommendations
-            """},
+            {"role": "system", "content": system_prompt},
             {"role": "user", "content": query}
         ]
         
         response = openai.ChatCompletion.create(
-            model="llama3-70b-8192",  # Corrected Groq model name
+            model="llama3-70b-8192",
             messages=messages,
-            temperature=0.7,
-            max_tokens=500
+            temperature=0.75,
+            max_tokens=600,
+            top_p=0.9
         )
         return response.choices[0].message["content"]
     except Exception as e:
-        return f"🔮 An error occurred: {str(e)} Please try again."
+        return f"🔴 Cosmic Interference! Please try again. Error: {str(e)}"
 
-def chatbot(name, dob, time_of_birth, place_of_birth, zodiac_sign, query, chat_history):
-    # Initialize chat history if None
-    if chat_history is None:
-        chat_history = []
+def handle_chat(name, dob, time_of_birth, place_of_birth, zodiac, query, chat_history):
+    # Validate required fields
+    if not all([name.strip(), dob.strip(), query.strip()]):
+        return chat_history, "⚠️ Please fill all required fields (Name, DOB, and Question)"
+    
+    # Initialize chat history
+    chat_history = chat_history or []
+    
+    # Add user message
+    chat_history.append((query, None))
     
     # Get AI response
-    bot_response = get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac_sign, query)
+    bot_response = get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac, query)
     
     # Update chat history
-    chat_history.append((f"{query}", f"{bot_response}"))
+    chat_history[-1] = (query, bot_response)
     
     return chat_history, chat_history
 
-# Create Gradio components
-with gr.Blocks(theme=gr.themes.Soft(), title="MysticAI Astrologer 🔮") as demo:
-    gr.Markdown("# 🌌 MysticAI Astrologer")
-    gr.Markdown("Discover your cosmic blueprint with AI-powered astrology insights")
+# Gradio UI
+with gr.Blocks(
+    theme=gr.themes.Soft(
+        primary_hue="purple",
+        secondary_hue="pink"
+    ),
+    title="Celestial Guide 🌙",
+    css=".gradio-container {background: url('https://example.com/space-bg.jpg')}"
+) as demo:
+    gr.Markdown("""
+    # 🌟 Celestial Guide - AI Astrology Companion
+    *Your personal cosmic advisor powered by Groq & Llama-3*
+    """)
     
     with gr.Row():
-        with gr.Column(scale=1):
-            name = gr.Textbox(label="Full Name")
-            dob = gr.Textbox(label="Date of Birth (DD-MM-YYYY)")
-            time_of_birth = gr.Textbox(label="Exact Birth Time (HH:MM AM/PM)")
-            place_of_birth = gr.Textbox(label="Birth Place (City, Country)")
+        with gr.Column(scale=1, min_width=300):
+            gr.Markdown("## 📝 Birth Details")
+            name = gr.Textbox(label="Full Name", placeholder="Enter your full name...")
+            dob = gr.Textbox(label="Date of Birth (DD-MM-YYYY)", placeholder="e.g., 15-04-1990")
+            time_of_birth = gr.Textbox(label="Birth Time (HH:MM AM/PM)", placeholder="e.g., 08:45 PM")
+            place_of_birth = gr.Textbox(label="Birth Place", placeholder="City, Country")
             zodiac = gr.Dropdown(
+                label="Sun Sign",
                 choices=["Aries", "Taurus", "Gemini", "Cancer", "Leo", "Virgo",
                         "Libra", "Scorpio", "Sagittarius", "Capricorn", "Aquarius", "Pisces"],
-                label="Sun Sign"
+                value="Aries"
             )
-        
+            
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(height=500)
-            query = gr.Textbox(label="Your Astrological Question", placeholder="Ask about your career, relationships, or upcoming transits...")
-            state = gr.State()
-            submit_btn = gr.Button("Consult the Stars ✨", variant="primary")
+            gr.Markdown("## 💬 Cosmic Consultation")
+            chatbot = gr.Chatbot(
+                height=600,
+                label="Astrology Chat",
+                avatar_images=("👤", "🔮"),
+                bubble_full_width=False
+            )
+            query = gr.Textbox(
+                label="Your Cosmic Question",
+                placeholder="Ask about love, career, or spiritual growth...",
+                lines=3
+            )
+            with gr.Row():
+                submit_btn = gr.Button("Consult the Stars 🌠", variant="primary")
+                clear_btn = gr.Button("New Session ♻️")
+    
+    state = gr.State()
     
+    # Event handlers
     submit_btn.click(
-        chatbot,
+        handle_chat,
         inputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, state],
         outputs=[chatbot, state]
     )
+    
+    query.submit(
+        handle_chat,
+        inputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, state],
+        outputs=[chatbot, state]
+    )
+    
+    clear_btn.click(lambda: (None, None, None, None, None, None, []), 
+                   outputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, chatbot])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False)