Update app.py

app.py

@@ -1,11 +1,18 @@
 import streamlit as st
-from groq import Groq
-from huggingface_hub import InferenceClient
-from langdetect import detect
 import os
+from transformers import pipeline
+from langdetect import detect
+from groq import Groq
+
+# Load Hugging Face token from environment
+HF_TOKEN = os.environ.get("homeo_doc")
+if not HF_TOKEN:
+    st.error("Missing Hugging Face API token. Set 'homeo_doc' in environment variables.")
 
-# Initialize clients
-hf_client = InferenceClient(token=os.environ.get("homeo_doc"))
+# Initialize translation pipeline
+translator = pipeline("translation", model="facebook/nllb-200-distilled-600M", token=HF_TOKEN)
+
+# Initialize Groq client for homeopathic advice
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 # Language code mapping for NLLB-200
@@ -23,26 +30,20 @@ def translate_text(text, target_lang='eng_Latn'):
     try:
         source_lang = detect(text)
         source_code = LANG_CODE_MAP.get(source_lang, 'eng_Latn')
-        
-        result = hf_client.translation(
-            text=text,
-            model="facebook/nllb-200-distilled-600M",
-            src_lang=source_code,
-            tgt_lang=target_lang
-        )
-        return result
+        translation = translator(text)[0]['translation_text']
+        return translation
     except Exception as e:
         st.error(f"Translation error: {str(e)}")
         return text
 
 def get_homeopathic_advice(symptoms):
-    """Get medical advice using Groq"""
+    """Get medical advice using Groq API"""
     try:
         response = groq_client.chat.completions.create(
             model="llama3-70b-8192",
             messages=[{
                 "role": "user",
-                "content": f"Act as homeopathic expert. Suggest remedies for: {symptoms}"
+                "content": f"Act as a homeopathic expert. Suggest remedies for: {symptoms}"
             }],
             temperature=0.3
         )
@@ -63,26 +64,25 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 if prompt := st.chat_input("Describe symptoms in any language"):
-    # User input
     st.session_state.messages.append({"role": "user", "content": prompt})
-    
+
     # Process input
     with st.spinner("Analyzing..."):
-        # Translate to English
-        english_input = translate_text(prompt, 'eng_Latn')
-        
+        # Translate input to English
+        english_input = translate_text(prompt)
+
         # Get medical advice
         english_advice = get_homeopathic_advice(english_input)
-        
+
         # Translate back to original language
         source_lang = detect(prompt)
-        translated_advice = translate_text(english_advice, LANG_CODE_MAP.get(source_lang, 'eng_Latn'))
-        
+        translated_advice = translate_text(english_advice)
+
         # Format response
         final_response = f"""
         **English Recommendation:**  
         {english_advice}
-
+        
         **Translated Recommendation ({source_lang.upper()}):**  
         {translated_advice}
         """
@@ -93,4 +93,4 @@ if prompt := st.chat_input("Describe symptoms in any language"):
     st.session_state.messages.append({"role": "assistant", "content": final_response})
 
 # Disclaimer
-st.caption("⚠️ This is not medical advice. Consult a professional.")
+st.caption("⚠️ This is not medical advice. Consult a professional.")