Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import os
 from groq import Groq
 from langdetect import detect, DetectorFactory
-import json
+import re
 
 # Ensure consistent language detection
 DetectorFactory.seed = 0
@@ -22,25 +22,28 @@ Your response must include:
 - Best possible remedy
 - Proper dosage and administration
 - Scientific or historical sources for reference
-- Store/availability of the medicine
 Ensure accuracy in medical details and maintain professionalism."""
 
+# Improve language detection, especially for Roman Urdu
+ROMAN_URDU_PATTERN = re.compile(r'\b(mujhe|kamar|dard|hai|aap|kyun|kaisi|bohot|zyada|masla|kar|raha|rahi|hun|ho|hai)\b', re.IGNORECASE)
+
 def detect_language(text):
-    """Detect the language of the input text."""
+    """Detect the language of the input text, with special handling for Roman Urdu."""
     try:
-        lang = detect(text)
-        return lang
+        if ROMAN_URDU_PATTERN.search(text):
+            return "ur"  # Detect Roman Urdu correctly
+        return detect(text)
     except:
         return "en"  # Default to English if detection fails
 
-def get_homeopathic_advice(symptoms, lang="en"):
+def get_homeopathic_advice(symptoms):
     """Search the web for remedies and generate a professional homeopathic response."""
     try:
         response = groq_client.chat.completions.create(
             model="llama3-70b-8192",
             messages=[
                 {"role": "system", "content": CHATBOT_ROLE},
-                {"role": "user", "content": f"Find the best homeopathic remedy for: {symptoms}. Provide remedy details, usage, dose, sources, and availability."}
+                {"role": "user", "content": f"Find the best homeopathic remedy for: {symptoms}. Provide remedy details, usage, dose, and sources."}
             ],
             temperature=0.3
         )
@@ -85,7 +88,7 @@ if prompt := st.chat_input("Describe your symptoms..."):
         detected_lang = detect_language(prompt)
 
         # Get homeopathic remedy
-        homeo_advice = get_homeopathic_advice(prompt, detected_lang)
+        homeo_advice = get_homeopathic_advice(prompt)
 
         # Translate advice if needed
         if detected_lang != "en":