Update app.py

app.py

@@ -1,151 +1,141 @@
+# app.py
 import streamlit as st
-import pandas as pd
 from groq import Groq
 import os
+import requests
+from bs4 import BeautifulSoup
+import re
+from urllib.parse import quote_plus
 
-# --------------------------
-# 1. Dataset Setup (Updated with Urdu)
-# --------------------------
-data = {
-    "symptoms_en": ["headache", "fever", "insomnia", "anxiety", "cough"],
-    "symptoms_es": ["dolor de cabeza", "fiebre", "insomnio", "ansiedad", "tos"],
-    "symptoms_hi": ["सिरदर्द", "बुखार", "अनिद्रा", "चिंता", "खांसी"],
-    "symptoms_ur": ["سر درد", "بخار", "بے خوابی", "اضطراب", "کھانسی"],
-    "remedy": ["Belladonna", "Aconitum", "Coffea", "Argentum nitricum", "Drosera"],
-    "potency": ["30C"]*5,
-    "usage": ["3 times daily"]*5
-}
-
-df = pd.DataFrame(data)
-
-# --------------------------
-# 2. Groq Setup (Multilingual Processing)
-# --------------------------
+# Groq API setup
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-def analyze_symptoms(text, language):
-    """Use Groq's Llama 3 for symptom analysis"""
-    lang_map = {
-        'en': 'English',
-        'es': 'Spanish', 
-        'hi': 'Hindi',
-        'ur': 'Urdu'
+# Web scraping functions
+def google_search(query):
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
     }
+    encoded_query = quote_plus(query)
+    url = f"https://www.google.com/search?q={encoded_query}&gl=us&hl=en"
+    
+    try:
+        response = requests.get(url, headers=headers)
+        soup = BeautifulSoup(response.text, 'html.parser')
+        results = []
+        
+        for g in soup.find_all('div', class_='tF2Cxc'):
+            link = g.find('a')['href']
+            title = g.find('h3').text
+            snippet = g.find('div', class_='VwiC3b')
+            if snippet:
+                results.append({
+                    'title': title,
+                    'link': link,
+                    'snippet': snippet.text
+                })
+        return results[:3]  # Return top 3 results
+    except Exception as e:
+        st.error(f"سرچ ایرر: {str(e)}")
+        return []
+
+# Homeopathic dosage extraction (Urdu supported)
+def extract_dosage_info(text):
+    patterns = [
+        r'(\d+[\s-]*\d*\s*(بار|قطرے|گولیاں|خوارج)\s*(فی دن|روزانہ))',
+        r'(دن میں \d+\s*بار)',
+        r'(\d+-\d+\s*گھنٹے کے وقفے سے)'
+    ]
     
-    prompt = f"""Task: Extract medical symptoms from this {lang_map[language]} text.
-    Text: {text}
-    Output format: Comma-separated symptoms in {lang_map[language]}.
-    Example: headache, muscle pain, dry cough"""
+    for pattern in patterns:
+        match = re.search(pattern, text)
+        if match:
+            return match.group()
+    return "دن میں 3 بار (ڈیفالٹ)"
+
+# Chatbot logic
+def homeo_chatbot_urdu(user_input):
+    # Step 1: Symptom extraction in Urdu
+    symptom_prompt = f"""صارف کی تفصیل: {user_input}
+    ہومیوپیتھک علامات کو انگلش میں کاما سے علیحدہ فہرست میں نکالیں۔
+    مثال: headache, dry cough, fever with chills"""
     
     try:
-        chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
+        symptom_response = client.chat.completions.create(
+            messages=[{"role": "user", "content": symptom_prompt}],
             model="llama3-70b-8192",
-            temperature=0.1
+            temperature=0.2
         )
-        return [s.strip() for s in chat_completion.choices[0].message.content.split(",")]
+        symptoms = symptom_response.choices[0].message.content.split(", ")
     except Exception as e:
-        st.error(f"Error in Groq API: {str(e)}")
-        return []
+        return f"ایرر: {str(e)}"
+
+    # Step 2: Web search for remedies
+    search_query = f"homeopathic remedies for {' '.join(symptoms)} site:.edu OR site:.gov"
+    search_results = google_search(search_query)
+    
+    remedies = []
+    for result in search_results:
+        remedies.append({
+            'title': result['title'],
+            'link': result['link'],
+            'content': result['snippet']
+        })
 
-# --------------------------
-# 3. Streamlit UI (Multilingual Interface)
-# --------------------------
-st.set_page_config(page_title="Homeo Doctor", page_icon="🌿")
+    # Step 3: Prepare response in Urdu
+    final_response = "🌿 **تجویز کردہ ہومیوپیتھک علاج:**\n\n"
+    for idx, remedy in enumerate(remedies, 1):
+        dosage = extract_dosage_info(remedy['content'])
+        final_response += f"""\
+**آپشن {idx}:**
+- **دوا:** {remedy['content'].split(' ')[0]}
+- **علامات:** {', '.join(symptoms[:3])}
+- **خوارک:** {dosage}
+- **طریقہ استعمال:** {"5 قطرے پانی میں" if 'قطرے' in dosage else "2 گولیاں"}
+- **دورانیہ:** {"3 دن" if 'حاد' in user_input else "1 ہفتہ"}
+- **ماخذ:** [{remedy['title']}]({remedy['link']})
 
-# Language selection
-language = st.sidebar.selectbox("भाषा चुनें/Select Language", 
-                              ['English', 'Español', 'हिन्दी', 'اردو'],
-                              format_func=lambda x: x)
+"""
+    
+    return final_response
 
-lang_code = {
-    'English': 'en',
-    'Español': 'es', 
-    'हिन्दी': 'hi',
-    'اردو': 'ur'
-}[language]
+# Streamlit UI in Urdu
+st.set_page_config(page_title="ہومیوپیتھک ڈاکٹر", page_icon="🌿")
 
-# Header with dynamic language
-headers = {
-    'en': "🌿 Homeopathic Doctor",
-    'es': "🌿 Doctor Homeopático", 
-    'hi': "🌿 होम्योपैथिक डॉक्टर",
-    'ur': "🌿 ہومیوپیتھک ڈاکٹر"
+# Urdu interface
+st.title("🌐 لائیو ہومیوپیتھک مشیر")
+st.markdown("""
+<style>
+[data-testid="stMarkdownContainer"] ul {
+    padding-right: 40px;
 }
-st.header(headers[lang_code])
+</style>
+""", unsafe_allow_html=True)
 
-# MCQ Symptoms
-symptom_col = f"symptoms_{lang_code}"
-symptoms = df[symptom_col].tolist()
-selected = st.multiselect(
-    label={
-        'en': "Select your symptoms:",
-        'es': "Seleccione sus síntomas:",
-        'hi': "अपने लक्षण चुनें:",
-        'ur': "اپنی علامات منتخب کریں:"
-    }[lang_code],
-    options=symptoms
-)
+# Chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        {"role": "assistant", "content": "السلام علیکم! آپ کی کیا علامات ہیں؟ مثالوں:\n- سر درد اور چکر آنا\n- کھانسی کے ساتھ بخار\n- پیٹ درد اور بھوک نہ لگنا"}
+    ]
 
-# Chat Input
-chat_input = st.text_input(
-    label={
-        'en': "Describe other symptoms:",
-        'es': "Describa otros síntomas:",
-        'hi': "अन्य लक्षण बताएं:",
-        'ur': "دیگر علامات بیان کریں:"
-    }[lang_code],
-    placeholder={
-        'en': "Type symptoms not listed above...",
-        'ur': "اوپر درج نہ کی گئی علامات ٹائپ کریں..."
-    }[lang_code] if lang_code == 'ur' else ""
-)
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
-# Diagnosis Logic
-if st.button({
-    'en': "Get Recommendation", 
-    'es': "Obtener Recomendación",
-    'hi': "सिफारिश प्राप्त करें",
-    'ur': "تجویز حاصل کریں"
-}[lang_code]):
-    # Process chat input
-    if chat_input:
-        analyzed_symptoms = analyze_symptoms(chat_input, lang_code)
-        selected += [s for s in analyzed_symptoms if s in symptoms]
-    
-    # Find best remedy
-    if not selected:
-        st.warning({
-            'en': "Please select or describe symptoms!",
-            'ur': "براہ کرم علامات منتخب کریں یا بیان کریں!"
-        }[lang_code] if lang_code == 'ur' else "")
-    else:
-        matches = df[df[symptom_col].isin(selected)]
-        if not matches.empty:
-            remedy = matches.iloc[0]
-            st.success(f"""
-                **{remedy['remedy']}**  
-                {{
-                    'en': "Potency: {remedy['potency']}",
-                    'ur': "طاقت: {remedy['potency']}"
-                }}[lang_code]  
-                {{
-                    'en': "Usage: {remedy['usage']}",
-                    'ur': "استعمال: {remedy['usage']}"
-                }}[lang_code]
-            """)
-        else:
-            st.error({
-                'en': "No matching remedy found. Consult a professional.",
-                'ur': "کوئی مماثل دوا نہیں ملی۔ پیشہ ور سے مشورہ کریں۔"
-            }[lang_code])
+if prompt := st.chat_input("اپنی علامات یہاں درج کریں..."):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
+        with st.spinner("تجویز تیار کی جا رہی ہے..."):
+            full_response = homeo_chatbot_urdu(prompt)
+            st.markdown(full_response)
+    st.session_state.messages.append({"role": "assistant", "content": full_response})
 
 # Disclaimer
 st.markdown("""
 ---
-**⚠️ Notice/نوٹس:**  
-This is not medical advice. Always consult a qualified homeopath.  
-यह चिकित्सीय सलाह नहीं है। हमेशा योग्य होम्योपैथ से सलाह लें۔  
-Esto no es un consejo médico. Siempre consulte a un homeópata calificado.  
-یہ طبی مشورہ نہیں ہے۔ ہمیشہ کوالیفائڈ ہومیوپیتھک ڈاکٹر سے مشورہ کریں۔
+**⚠️ نوٹس:**  
+یہ طبی مشورہ نہیں ہے۔ ہمیشہ کوالیفائیڈ ڈاکٹر سے رجوع کریں۔  
+This is not medical advice. Always consult a qualified practitioner.
 """)