Update app.py

app.py

@@ -2,76 +2,76 @@ import gradio as gr
 import openai
 import os
 
-# Configure OpenAI API
+# Configure OpenAI API for Groq
 openai.api_key = os.getenv("API_KEY")
 openai.api_base = "https://api.groq.com/openai/v1"
 
-# Function to get AI-generated astrology response
 def get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac_sign, query):
     try:
-        # Construct the message
         messages = [
-            {"role": "system", "content": """
-                You are a mystical astrologer. Provide detailed insights about zodiac signs, planetary alignments, 
-                and astrological predictions based on the user's inputs. Use a mystical and engaging tone.
+            {"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": "user", "content": f"""
-                Name: {name}
-                Date of Birth: {dob}
-                Time of Birth: {time_of_birth}
-                Place of Birth: {place_of_birth}
-                Zodiac Sign: {zodiac_sign}
-                Query: {query}
-            """}
+            {"role": "user", "content": query}
         ]
-        # Generate response from the AI
+        
         response = openai.ChatCompletion.create(
-            model="llama-3.3-70b-versatile",
-            messages=messages
+            model="llama3-70b-8192",  # Corrected Groq model name
+            messages=messages,
+            temperature=0.7,
+            max_tokens=500
         )
         return response.choices[0].message["content"]
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"🔮 An error occurred: {str(e)} Please try again."
 
-# Chatbot function to handle user input and history
-def chatbot(name, dob, time_of_birth, place_of_birth, zodiac_sign, query, history=[]):
-    # Get the response from the AI
+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 = []
+    
+    # Get AI response
     bot_response = get_groq_response(name, dob, time_of_birth, place_of_birth, zodiac_sign, query)
-    # Append the conversation to history
-    history.append((f"You: {query}", f"Astrologer: {bot_response}"))
-    return history, history
+    
+    # Update chat history
+    chat_history.append((f"{query}", f"{bot_response}"))
+    
+    return chat_history, chat_history
 
-# Define Gradio interface
-chat_interface = gr.Interface(
-    fn=chatbot,  # Function to call for interaction
-    inputs=[
-        gr.Textbox(label="Your Name"),  # Name input
-        gr.Textbox(label="Date of Birth (e.g., 01-01-2000)"),  # DOB input
-        gr.Textbox(label="Time of Birth (e.g., 10:30 AM)"),  # Time of Birth input
-        gr.Textbox(label="Place of Birth"),  # Place of Birth input
-        gr.Dropdown(
-            label="Your Zodiac Sign",
-            choices=[
-                "Aries", "Taurus", "Gemini", "Cancer", "Leo", "Virgo",
-                "Libra", "Scorpio", "Sagittarius", "Capricorn", "Aquarius", "Pisces"
-            ]
-        ),  # Dropdown for zodiac sign
-        gr.Textbox(label="Your Question for the Astrologer"),  # Query input
-        gr.State(),  # State for chat history
-    ],
-    outputs=[
-        gr.Chatbot(label="Astrology Chat"),  # Chatbot interface for responses
-        gr.State(),  # State to maintain chat history
-    ],
-    title="Astrology AI Assistant 🌌",
-    description=(
-        "Welcome to your personal Astrology AI Assistant! "
-        "Ask questions about your zodiac sign, horoscope, and planetary alignments. "
-        "Enter your details and let the stars guide you. ✨"
-    ),
-    theme="dark"  # Optional: Gradio theme customization
-)
+# 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")
+    
+    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)")
+            zodiac = gr.Dropdown(
+                choices=["Aries", "Taurus", "Gemini", "Cancer", "Leo", "Virgo",
+                        "Libra", "Scorpio", "Sagittarius", "Capricorn", "Aquarius", "Pisces"],
+                label="Sun Sign"
+            )
+        
+        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")
+    
+    submit_btn.click(
+        chatbot,
+        inputs=[name, dob, time_of_birth, place_of_birth, zodiac, query, state],
+        outputs=[chatbot, state]
+    )
 
-# Launch the Gradio app
 if __name__ == "__main__":
-    chat_interface.launch()
+    demo.launch()