Update app.py

app.py

@@ -111,6 +111,146 @@ st.markdown("""
 </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}],
+            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 = [