Update app.py

app.py

@@ -1,36 +1,26 @@
 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")
+openai.api_key = os.getenv("GROQ_API_KEY")
 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:
+        system_prompt = f"""You are Master Celestia, an AI astrologer. Analyze this birth chart:
         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."""
-
+        Provide insights about:
+        1. Current planetary transits
+        2. Career and relationships
+        3. Personalized recommendations
+        """
+        
         messages = [
             {"role": "system", "content": system_prompt},
             {"role": "user", "content": query}
@@ -39,95 +29,65 @@ def get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac_sign, que
         response = openai.ChatCompletion.create(
             model="llama3-70b-8192",
             messages=messages,
-            temperature=0.75,
-            max_tokens=600,
-            top_p=0.9
+            temperature=0.7,
+            max_tokens=500
         )
         return response.choices[0].message["content"]
     except Exception as e:
-        return f"🔴 Cosmic Interference! Please try again. Error: {str(e)}"
+        return f"Error: {str(e)}"
 
+# Renamed function to avoid conflict
 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, query)
-    
-    # Update chat history
-    chat_history[-1] = (query, bot_response)
+    if query:
+        # Get AI response
+        bot_response = get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac, query)
+        chat_history.append((query, bot_response))
     
     return chat_history, chat_history
 
-# 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.Blocks(theme=gr.themes.Soft(), title="Astro Guide") as demo:
+    gr.Markdown("# 🌟 Astro Guide - AI Astrologer")
     
     with gr.Row():
-        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")
+        with gr.Column(scale=1):
+            name = gr.Textbox(label="Full Name")
+            dob = gr.Textbox(label="DOB (DD-MM-YYYY)")
+            time_of_birth = gr.Textbox(label="Birth Time")
+            place_of_birth = gr.Textbox(label="Birth Place")
             zodiac = gr.Dropdown(
-                label="Sun Sign",
                 choices=["Aries", "Taurus", "Gemini", "Cancer", "Leo", "Virgo",
                         "Libra", "Scorpio", "Sagittarius", "Capricorn", "Aquarius", "Pisces"],
-                value="Aries"
+                label="Sun Sign"
             )
-            
+        
         with gr.Column(scale=2):
-            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
+            # Renamed chatbot component
+            chat_interface = gr.Chatbot(height=500)
+            query = gr.Textbox(label="Your Question")
+            state = gr.State()
+            submit_btn = gr.Button("Ask the Stars", variant="primary")
+            clear_btn = gr.Button("Clear")
+
+    # Fixed click handlers
     submit_btn.click(
-        handle_chat,
+        handle_chat,  # Using the renamed function
         inputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, state],
-        outputs=[chatbot, state]
+        outputs=[chat_interface, state]
     )
     
     query.submit(
         handle_chat,
         inputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, state],
-        outputs=[chatbot, state]
+        outputs=[chat_interface, state]
     )
     
-    clear_btn.click(lambda: (None, None, None, None, None, None, []), 
-                   outputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, chatbot])
+    clear_btn.click(
+        lambda: (None, None, None, None, None, None, []),
+        outputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, chat_interface],
+        show_progress=False
+    )
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch()