Update app.py

app.py

@@ -1,61 +1,90 @@
 from flask import Flask, request, jsonify
 from flask_cors import CORS
-from deep_translator import GoogleTranslator
 import json
+from deep_translator import GoogleTranslator
 
 app = Flask(__name__)
 CORS(app)
 
+# Load disease data
 with open("diseases.json", "r", encoding="utf-8") as f:
     disease_data = json.load(f)
 
+DISEASE_ALIASES = {
+    "ringworm": "Tinea Ringworm Candidiasis",
+    "fungal infection": "Tinea Ringworm Candidiasis",
+    "eczema": "eczema",
+    "warts": "Warts Molluscum Viral Infections",
+    "melanoma": "melanoma",
+    "nevus": "nevus"
+}
+
+SUPPORTED_LANGUAGES = {
+    "en": "English",
+    "ta": "Tamil",
+    "hi": "Hindi",
+    "te": "Telugu",
+    "ml": "Malayalam",
+    "kn": "Kannada",
+    "bn": "Bengali",
+    "mr": "Marathi",
+    "ur": "Urdu",
+    "es": "Spanish"
+}
+
+@app.route("/")
+def home():
+    return "Chatbot backend running"
+
 @app.route("/chat", methods=["POST"])
 def chat():
     data = request.json
 
-    message = data.get("message", "")
-    language = data.get("language", "en")
+    user_input = data.get("message", "")
+    user_lang = data.get("language", "en")  # 🔥 THIS IS KEY
 
-    print("LANG RECEIVED:", language)  # 👈 DEBUG
+    print("LANG RECEIVED 👉", user_lang)   # 🔥 DEBUG (HF logs)
 
-    disease = None
-    for d in disease_data:
-        if d.lower() in message.lower():
-            disease = d
-            break
+    if user_lang not in SUPPORTED_LANGUAGES:
+        user_lang = "en"
 
-    if not disease:
-        return jsonify({"reply": "Disease not found"})
-
-    info = disease_data[disease]
+    # Translate input to English
+    if user_lang != "en":
+        translated_input = GoogleTranslator(
+            source=user_lang, target="en"
+        ).translate(user_input)
+    else:
+        translated_input = user_input
 
-    english_reply = f"""
-Disease: {disease}
+    normalized_input = translated_input.lower()
 
-Description:
-{info['description']}
+    matched_disease = None
+    for key, value in DISEASE_ALIASES.items():
+        if key in normalized_input:
+            matched_disease = value
+            break
 
-Symptoms:
-- """ + "\n- ".join(info["symptoms"]) + """
+    if not matched_disease or matched_disease not in disease_data:
+        return jsonify({"reply": "Disease information not found"})
 
-Precautions:
-- """ + "\n- ".join(info["precautions"]) + """
+    info = disease_data[matched_disease]
 
-⚠ Consult dermatologist
-"""
+    english_response = (
+        f"Disease: {matched_disease}\n\n"
+        f"Description:\n{info['description']}\n\n"
+        f"Symptoms:\n➤ " + "\n➤ ".join(info["symptoms"]) + "\n\n"
+        f"Precautions:\n➤ " + "\n➤ ".join(info["precautions"]) + "\n\n"
+        "⚠ Educational purpose only"
+    )
 
-    if language != "en":
-        reply = GoogleTranslator(
-            source="en", target=language
-        ).translate(english_reply)
+    if user_lang != "en":
+        final_response = GoogleTranslator(
+            source="en", target=user_lang
+        ).translate(english_response)
     else:
-        reply = english_reply
+        final_response = english_response
 
-    return jsonify({"reply": reply})
-
-@app.route("/")
-def home():
-    return "Chatbot running"
+    return jsonify({"reply": final_response})
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)