Update app.py

app.py

@@ -3,8 +3,8 @@ import os
 import requests
 
 # Mistral API setup
-MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY", "yQdfM8MLbX9uhInQ7id4iUTwN4h4pDLX")
-MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions "
+MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY", "yQdfM8MLbX9uhInQ7id4iUTwN4h4pDLX")  # Replace with your actual API key
+MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"  # Removed the trailing space
 
 def query_mistral(prompt, max_tokens=1024, temperature=0.7):
     """Query Mistral AI API"""
@@ -22,71 +22,64 @@ def query_mistral(prompt, max_tokens=1024, temperature=0.7):
 
     try:
         response = requests.post(MISTRAL_API_URL, headers=headers, json=payload, timeout=10)
-        response.raise_for_status()  # Raises exception for HTTP errors
+        response.raise_for_status()
         return response.json()['choices'][0]['message']['content']
     except requests.exceptions.RequestException as e:
-        return f"[Error] Failed to connect to Mistral API: {str(e)}"
+        return f"Sorry, I'm having trouble connecting to my knowledge base. Please try again later. Error: {str(e)}"
     except Exception as e:
-        return f"[Error] Unexpected error: {str(e)}"
+        return f"An unexpected error occurred. Please rephrase your question or try again later. Error: {str(e)}"
 
-def analyze_symptoms(age, gender, symptoms, duration, severity):
-    """Analyze symptoms using Mistral AI"""
-    prompt = f"""As a medical professional, analyze these symptoms:
-    - Patient: {age}-year-old {gender}
-    - Symptoms: {symptoms}
-    - Duration: {duration}
-    - Severity: {severity}/10
-
-    Provide:
-    1. Possible conditions (list 2-3 most likely)
-    2. Recommended next steps
-    3. When to seek immediate care
-
-    Keep response under 200 words and in simple language."""
+def personalize_response(name, message):
+    """Personalize the chatbot response"""
+    prompt = f"""You are a friendly and empathetic personal medical assistant named Dr. Alex. 
+    The user {name} asked: "{message}"
+    
+    Respond with:
+    1. A caring, personalized greeting using the user's name
+    2. Helpful medical information (but no definitive diagnoses)
+    3. Follow-up questions if more information is needed
+    4. Clear recommendation on when to see a real doctor
+    5. A warm closing
     
+    Keep the response under 200 words, conversational, and supportive.
+    """
     return query_mistral(prompt)
 
-def chatbot_response(message, history):
-    """Doctor chatbot response function"""
+def chatbot_response(message, history, name):
+    """Personal medical assistant response function"""
     message_lower = message.lower()
-
+    
+    # Simple greetings
     if any(word in message_lower for word in ['hello', 'hi', 'hey']):
-        return "Hello! I'm Doctor Bot. Would you like to discuss symptoms or health concerns?"
+        return f"Hello {name}! I'm Dr. Alex, your personal medical assistant. How can I help you today?"
     elif any(word in message_lower for word in ['thank', 'thanks']):
-        return "You're welcome! How else can I assist you today?"
-
-    # Use Mistral for medical questions
-    prompt = f"""You are a helpful medical chatbot. The user asked:
-    "{message}"
-
-    Respond with:
-    1. Helpful medical information (but no diagnoses)
-    2. Questions to clarify symptoms if needed
-    3. Recommendation to see a doctor when appropriate
-
-    Keep response under 150 words and friendly."""
-
-    return query_mistral(prompt)
+        return f"You're very welcome, {name}! I'm happy to help. Is there anything else you'd like to discuss?"
+    
+    # Personalize medical responses
+    return personalize_response(name, message)
 
 # Gradio Interface
-with gr.Blocks(title="Medical Chatbot", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🩺 AI Medical Assistant")
+with gr.Blocks(title="Personal Medical Assistant", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 👨‍⚕️ Your Personal Medical Assistant")
+    
+    # Get user's name for personalization
+    name = gr.Textbox(label="Your Name", placeholder="Please enter your name for personalized care")
     
     with gr.Tabs():
-        with gr.TabItem("Chat with Doctor Bot"):
-            gr.Markdown("## Ask general health questions")
+        with gr.TabItem("Chat with Dr. Alex"):
+            gr.Markdown("## Ask me anything about your health")
             chatbot = gr.ChatInterface(
-                fn=chatbot_response,
+                fn=lambda msg, hist: chatbot_response(msg, hist, name.value),
                 examples=[
-                    "What should I do for a fever?",
-                    "How to treat a sprained ankle?",
-                    "When should I worry about a headache?"
+                    "I've had a headache for 2 days",
+                    "What are symptoms of dehydration?",
+                    "How can I improve my sleep?"
                 ],
-                title="AI Doctor Chat"
+                title="Dr. Alex - Your Personal Health Assistant"
             )
         
-        with gr.TabItem("Symptom Checker"):
-            gr.Markdown("## Complete this survey for symptom analysis")
+        with gr.TabItem("Symptom Analysis"):
+            gr.Markdown(f"## {name.value}, let's analyze your symptoms")
             with gr.Row():
                 with gr.Column():
                     age = gr.Number(label="Your Age", minimum=1, maximum=120)
@@ -97,12 +90,15 @@ with gr.Blocks(title="Medical Chatbot", theme=gr.themes.Soft()) as demo:
                         label="How long have you had these symptoms?"
                     )
                     severity = gr.Slider(1, 10, step=1, label="Severity (1-10)")
-                    analyze_btn = gr.Button("Analyze Symptoms", variant="primary")
+                    analyze_btn = gr.Button("Analyze My Symptoms", variant="primary")
                 with gr.Column():
-                    analysis_output = gr.Textbox(label="AI Analysis", lines=10, interactive=False)
+                    analysis_output = gr.Textbox(label="Dr. Alex's Analysis", lines=10, interactive=False)
             
             analyze_btn.click(
-                analyze_symptoms,
+                lambda age, gender, symptoms, duration, severity: personalize_response(
+                    name.value,
+                    f"As a {age}-year-old {gender}, I've had these symptoms: {symptoms} for {duration} with severity {severity}/10. What could this be?"
+                ),
                 inputs=[age, gender, symptoms, duration, severity],
                 outputs=analysis_output
             )