Update app.py

app.py

@@ -1,53 +1,97 @@
-# app.py (Updated Version)
-def multilingual_chatbot(user_input):
+# app.py
+import streamlit as st
+from groq import Groq
+import os
+import requests
+from bs4 import BeautifulSoup
+from langdetect import detect
+from urllib.parse import quote_plus
+
+# Initialize Groq client
+try:
+    client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+except Exception as e:
+    st.error(f"Groq initialization failed: {str(e)}")
+
+# Web scraping with error handling
+def google_search(query):
     try:
-        # Detect input language
+        headers = {"User-Agent": "Mozilla/5.0"}
+        encoded_query = quote_plus(query)
+        url = f"https://www.google.com/search?q={encoded_query}&gl=us&hl=en"
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        
+        soup = BeautifulSoup(response.text, 'html.parser')
+        return [
+            {
+                'title': result.find('h3').text,
+                'link': result.find('a')['href'],
+                'snippet': result.find('div', class_='VwiC3b').text
+            }
+            for result in soup.find_all('div', class_='tF2Cxc')[:3]
+        ]
+    except Exception as e:
+        st.error(f"Google search failed: {str(e)}")
+        return []
+
+# Main chatbot function
+def get_homeopathic_advice(user_input):
+    try:
+        # Language detection
         lang = detect(user_input)
         
-        # Step 1: Symptom extraction with homeopathy focus
-        symptom_prompt = f"""Extract ONLY homeopathic-relevant symptoms from this text as English comma-separated list:
+        # Step 1: Symptom extraction
+        symptom_prompt = f"""Extract homeopathic symptoms from this text as comma-separated list:
         {user_input}
-        Example Output: stinging pain, anxiety worse evenings, thirstlessness"""
+        Example: headache, dry cough, anxiety"""
         
-        symptom_response = client.chat.completions.create(
+        symptoms = client.chat.completions.create(
             messages=[{"role": "user", "content": symptom_prompt}],
-            model="llama3-70b-8192",
-            temperature=0.1
-        )
-        symptoms = symptom_response.choices[0].message.content.split(", ")
+            model="llama3-70b-8192"
+        ).choices[0].message.content.split(", ")
         
-        # Step 2: Focused homeopathic web search
-        search_query = (
+        # Step 2: Homeopathic-specific search
+        search_results = google_search(
             f"homeopathic remedy for {' '.join(symptoms)} "
-            f"(site:.edu OR site:.gov) "
-            f"-allopathic -conventional -pharmaceutical"
+            f"site:.edu OR site:.gov -allopathic"
         )
-        results = google_search(search_query)
         
-        # Filter homeopathic-specific results
-        homeo_keywords = ['homeopath', 'remedy', 'potency', 'materia medica']
-        filtered_results = [
-            r for r in results
-            if any(kw in r['snippet'].lower() for kw in homeo_keywords)
-        ]
-        
-        # Step 3: Generate strict homeopathic response
-        response_prompt = f"""You are a homeopathic doctor. Recommend ONLY homeopathic medicines in {lang} for these symptoms: {', '.join(symptoms)}.
-        Use this research data: {[r['snippet'] for r in filtered_results]}
-        Include:
-        - Medicine name in English
-        - Potency (like 30C, 200CK)
-        - Administration method
-        - Key symptoms it addresses
-        - Source reference"""
+        # Step 3: Generate response
+        response_prompt = f"""You're a homeopathic expert. Suggest remedies in {lang} for: {', '.join(symptoms)}
+        Use only this data: {[r['snippet'] for r in search_results]}
+        Include medicine name (English), potency, dosage and source."""
         
-        final_response = client.chat.completions.create(
+        return client.chat.completions.create(
             messages=[{"role": "user", "content": response_prompt}],
-            model="llama3-70b-8192",
-            temperature=0.3
-        )
-        
-        return final_response.choices[0].message.content
+            model="llama3-70b-8192"
+        ).choices[0].message.content
         
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}"
+
+# Streamlit UI Setup
+st.set_page_config(page_title="Homeo Advisor", page_icon="🌿")
+st.title("🌿 Homeopathic Chat Advisor")
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    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"])
+
+# User input handling
+if prompt := st.chat_input("Enter symptoms..."):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    
+    with st.chat_message("assistant"):
+        with st.spinner("Analyzing..."):
+            response = get_homeopathic_advice(prompt)
+            st.markdown(response)
+    st.session_state.messages.append({"role": "assistant", "content": response})
+
+# Disclaimer
+st.caption("⚠️ This is not medical advice. Consult a professional.")