Update app.py

app.py

@@ -38,39 +38,37 @@ def generate_doctor_response(history):
         yield history
         return
 
-    # Enhanced Medical Prompt - Comprehensive Doctor Approach
-    prompt = f"""You are an experienced and compassionate medical doctor providing a comprehensive consultation.
+    # Enhanced Medical Prompt - Conversational Doctor Style
+    prompt = f"""You are a friendly, experienced doctor having a natural conversation with a patient.
 
-Based on the patient's concern, provide a detailed response that includes:
+Respond in a warm, conversational way covering:
+- What might be causing their symptoms (keep it simple)
+- Specific medications they can take (with dosages)
+- Foods and drinks that will help
+- Lifestyle tips and rest recommendations
+- When they should see a doctor in person
 
-1. **Assessment**: Acknowledge their symptoms and provide initial medical assessment
-2. **Possible Causes**: Explain potential causes or conditions
-3. **Medications**: Recommend appropriate over-the-counter or prescription medications (with dosages when relevant)
-4. **Nutrition & Diet**: Suggest specific foods, nutrients, or dietary changes that can help
-5. **Lifestyle Modifications**: Recommend lifestyle changes, exercises, rest, or habits to adopt
-6. **Follow-up**: Advise when to see a doctor or what warning signs to watch for
+Important rules:
+- Write like you're talking to a friend, not writing a formal report
+- NO labels like "Doctor:", "Patient:", "Assessment:", "Medications:" etc.
+- Keep it natural and flowing, not structured lists
+- Be warm and reassuring
+- Keep response under 200 words
+- End with a simple disclaimer
 
-Guidelines:
-- Do NOT use labels like "Doctor:" or "Patient:" in your response
-- Be professional, empathetic, and thorough like a real doctor
-- Provide specific, actionable recommendations
-- Use medical terminology but explain it simply
-- Structure your response clearly with the categories above
-- End with: "⚕️ *Please consult a healthcare provider for proper diagnosis and personalized treatment plan.*"
+Patient says: {user_message}
 
-Patient's concern: {user_message}
-
-Comprehensive Medical Response:"""
+Your response:"""
 
     # Tokenize input
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=2048).to(model.device)
 
     gen_config = GenerationConfig(
-        temperature=0.7,
-        top_p=0.92,
-        top_k=50,
+        temperature=0.8,
+        top_p=0.95,
+        top_k=40,
         do_sample=True,
-        max_new_tokens=600,
+        max_new_tokens=300,
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
         repetition_penalty=1.18,
@@ -101,34 +99,61 @@ Comprehensive Medical Response:"""
 
 
 def clean_medical_response(response: str) -> str:
-    """Clean and format the medical response."""
+    """Clean and format the medical response for natural conversation."""
     # Remove common prefixes
-    prefixes = ["assistant:", "doctor:", "response:", "comprehensive medical response:", "medical response:"]
+    prefixes = ["assistant:", "doctor:", "response:", "comprehensive medical response:", 
+                "medical response:", "your response:", "dear patient"]
     response_lower = response.lower()
     for prefix in prefixes:
         if response_lower.startswith(prefix):
             response = response[len(prefix):].strip()
+            response = response.lstrip(',').strip()
             break
     
-    # Remove any remaining role labels
+    # Remove section headers/labels
     lines = response.split('\n')
     cleaned_lines = []
+    skip_patterns = ['doctor:', 'assistant:', 'patient:', 'assessment:', 'medications:', 
+                    'nutrition:', 'lifestyle:', 'follow-up:', '1.', '2.', '3.', '4.', '5.']
+    
     for line in lines:
-        if not line.lower().strip().startswith(('doctor:', 'assistant:', 'patient:')):
+        line_lower = line.lower().strip()
+        should_skip = False
+        for pattern in skip_patterns:
+            if line_lower.startswith(pattern):
+                should_skip = True
+                break
+        if not should_skip and line.strip():
             cleaned_lines.append(line)
+    
     response = '\n'.join(cleaned_lines)
     
-    # Ensure proper ending
-    if response and response[-1] not in '.!?':
-        response += '.'
+    # Stop at repetitive text or gibberish
+    if 'accordingly' in response.lower() or 'respectively' in response.lower():
+        sentences = response.split('.')
+        good_sentences = []
+        for sent in sentences:
+            if 'accordingly' not in sent.lower() and 'respectively' not in sent.lower():
+                good_sentences.append(sent)
+            else:
+                break
+        response = '. '.join(good_sentences)
+    
+    # Limit to reasonable length
+    sentences = [s.strip() for s in response.split('.') if s.strip()]
+    if len(sentences) > 8:
+        response = '. '.join(sentences[:8]) + '.'
+    else:
+        response = '. '.join(sentences) + '.'
     
-    # Add disclaimer if not present
-    if '⚕️' not in response and 'consult' not in response.lower():
-        response += '\n\n⚕️ *Please consult a healthcare provider for proper diagnosis and personalized treatment plan.*'
+    # Add disclaimer if not present and response is substantial
+    if len(response) > 50:
+        if '⚕️' not in response and 'consult' not in response.lower():
+            response += '\n\n⚕️ *If symptoms worsen or persist beyond 3-4 days, please see a doctor.*'
     
     # Fallback for very short responses
-    if len(response.strip()) < 30:
-        response = "I understand your concern. To provide you with comprehensive medical guidance including medications, diet, and lifestyle recommendations, could you please describe your symptoms in more detail? For example, when did they start, how severe are they, and have you noticed any triggers?"
+    if len(response.strip()) < 40:
+        response = "I understand you're not feeling well. Could you tell me a bit more about your symptoms? How long have you had them, and have you taken anything for them yet?"
     
     return response.strip()
 
@@ -149,8 +174,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
     
     gr.HTML("""
         <div class="medical-header">
-            <h1>🏥 AI Medical Doctor Consultation</h1>
-            <p>Comprehensive Medical Guidance • Medications • Nutrition • Lifestyle</p>
+            <h1>🏥 AI Doctor Chat</h1>
+            <p>Friendly Medical Advice • Like Talking to Your Doctor</p>
         </div>
     """)