Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 from bs4 import BeautifulSoup
 import re
 from urllib.parse import quote_plus
+from langdetect import detect
 
 # Groq API setup
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
@@ -33,249 +34,76 @@ def google_search(query):
                     'link': link,
                     'snippet': snippet.text
                 })
-        return results[:3]  # Return top 3 results
+        return results[:3]
     except Exception as e:
-        st.error(f"سرچ ایرر: {str(e)}")
+        st.error(f"Search Error: {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*گھنٹے کے وقفے سے)'
-    ]
-    
-    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"""
-    
+# Chatbot processing
+def multilingual_chatbot(user_input):
     try:
+        # Detect input language
+        lang = detect(user_input)
+        
+        # Step 1: Symptom extraction (English for searching)
+        symptom_prompt = f"""Extract medical symptoms from this text in English comma-separated format:
+        {user_input}
+        Example Output: headache, dry cough, fever"""
+        
         symptom_response = client.chat.completions.create(
             messages=[{"role": "user", "content": symptom_prompt}],
             model="llama3-70b-8192",
             temperature=0.2
         )
         symptoms = symptom_response.choices[0].message.content.split(", ")
-    except Exception as e:
-        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']
-        })
-
-    # 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']})
-
-"""
-    
-    return final_response
-
-# Streamlit UI in Urdu
-st.set_page_config(page_title="ہومیوپیتھک ڈاکٹر", page_icon="🌿")
-
-# Urdu interface
-st.title("🌐 لائیو ہومیوپیتھک مشیر")
-st.markdown("""
-<style>
-[data-testid="stMarkdownContainer"] ul {
-    padding-right: 40px;
-}
-</style>
-""", unsafe_allow_html=True)
-
-# Chat history
-if "messages" not in st.session_state:
-    st.session_state.messages = [
-        {"role": "assistant", "content": "السلام علیکم! آپ کی کیا علامات ہیں؟ مثالوں:\n- سر درد اور چکر آنا\n- کھانسی کے ساتھ بخار\n- پیٹ درد اور بھوک نہ لگنا"}
-    ]
-
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
-
-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("""
----
-**⚠️ نوٹس:**  
-یہ طبی مشورہ نہیں ہے۔ ہمیشہ کوالیفائیڈ ڈاکٹر سے رجوع کریں۔  
-This is not medical advice. Always consult a qualified practitioner.
-""")# app.py
-import streamlit as st
-from groq import Groq
-import os
-import requests
-from bs4 import BeautifulSoup
-import re
-from urllib.parse import quote_plus
-
-# Groq API setup
-client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
-
-# 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*گھنٹے کے وقفے سے)'
-    ]
-    
-    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:
-        symptom_response = client.chat.completions.create(
-            messages=[{"role": "user", "content": symptom_prompt}],
+        # Step 2: Web search
+        search_query = f"homeopathic remedies for {' '.join(symptoms)} site:.edu OR site:.gov"
+        results = google_search(search_query)
+        
+        # Step 3: Generate multilingual response
+        response_prompt = f"""Translate this medical advice to {lang} while keeping medicine names in English:
+        Suggested remedies for {', '.join(symptoms)}:
+        {[r['snippet'] for r in results]}
+        Include dosage instructions and administration method."""
+        
+        final_response = client.chat.completions.create(
+            messages=[{"role": "user", "content": response_prompt}],
             model="llama3-70b-8192",
-            temperature=0.2
+            temperature=0.3
         )
-        symptoms = symptom_response.choices[0].message.content.split(", ")
+        
+        return final_response.choices[0].message.content
+        
     except Exception as e:
-        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']
-        })
-
-    # 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']})
-
-"""
-    
-    return final_response
-
-# Streamlit UI in Urdu
-st.set_page_config(page_title="ہومیوپیتھک ڈاکٹر", page_icon="🌿")
+        return f"Error: {str(e)}"
 
-# Urdu interface
-st.title("🌐 لائیو ہومیوپیتھک مشیر")
-st.markdown("""
-<style>
-[data-testid="stMarkdownContainer"] ul {
-    padding-right: 40px;
-}
-</style>
-""", unsafe_allow_html=True)
+# Streamlit UI
+st.set_page_config(page_title="Homeo Assistant", page_icon="🌿")
+st.title("🌐 Multilingual Homeopathy Advisor")
 
-# Chat history
+# Initialize chat history
 if "messages" not in st.session_state:
-    st.session_state.messages = [
-        {"role": "assistant", "content": "السلام علیکم! آپ کی کیا علامات ہیں؟ مثالوں:\n- سر درد اور چکر آنا\n- کھانسی کے ساتھ بخار\n- پیٹ درد اور بھوک نہ لگنا"}
-    ]
+    st.session_state.messages = [{"role": "assistant", "content": "Describe your symptoms in any language"}]
 
+# Display chat messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-if prompt := st.chat_input("اپنی علامات یہاں درج کریں..."):
+# Chat input
+if prompt := st.chat_input("Type your symptoms..."):
     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})
+        with st.spinner("Analyzing symptoms..."):
+            response = multilingual_chatbot(prompt)
+            st.markdown(response)
+    st.session_state.messages.append({"role": "assistant", "content": response})
 
 # Disclaimer
 st.markdown("""
 ---
-**⚠️ نوٹس:**  
-یہ طبی مشورہ نہیں ہے۔ ہمیشہ کوالیفائیڈ ڈاکٹر سے رجوع کریں۔  
+**⚠️ Disclaimer:**  
 This is not medical advice. Always consult a qualified practitioner.
 """)