Update app.py

app.py

@@ -1,90 +1,61 @@
 from flask import Flask, request, jsonify
 from flask_cors import CORS
-import json
 from deep_translator import GoogleTranslator
+import json
 
 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
 
-    user_input = data.get("message", "")
-    user_lang = data.get("language", "en")  # 🔥 THIS IS KEY
+    message = data.get("message", "")
+    language = data.get("language", "en")
 
-    print("LANG RECEIVED 👉", user_lang)   # 🔥 DEBUG (HF logs)
+    print("LANG RECEIVED:", language)  # 👈 DEBUG
 
-    if user_lang not in SUPPORTED_LANGUAGES:
-        user_lang = "en"
+    disease = None
+    for d in disease_data:
+        if d.lower() in message.lower():
+            disease = d
+            break
 
-    # Translate input to English
-    if user_lang != "en":
-        translated_input = GoogleTranslator(
-            source=user_lang, target="en"
-        ).translate(user_input)
-    else:
-        translated_input = user_input
+    if not disease:
+        return jsonify({"reply": "Disease not found"})
 
-    normalized_input = translated_input.lower()
+    info = disease_data[disease]
 
-    matched_disease = None
-    for key, value in DISEASE_ALIASES.items():
-        if key in normalized_input:
-            matched_disease = value
-            break
+    english_reply = f"""
+Disease: {disease}
+
+Description:
+{info['description']}
 
-    if not matched_disease or matched_disease not in disease_data:
-        return jsonify({"reply": "Disease information not found"})
+Symptoms:
+- """ + "\n- ".join(info["symptoms"]) + """
 
-    info = disease_data[matched_disease]
+Precautions:
+- """ + "\n- ".join(info["precautions"]) + """
 
-    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"
-    )
+⚠ Consult dermatologist
+"""
 
-    if user_lang != "en":
-        final_response = GoogleTranslator(
-            source="en", target=user_lang
-        ).translate(english_response)
+    if language != "en":
+        reply = GoogleTranslator(
+            source="en", target=language
+        ).translate(english_reply)
     else:
-        final_response = english_response
+        reply = english_reply
 
-    return jsonify({"reply": final_response})
+    return jsonify({"reply": reply})
+
+@app.route("/")
+def home():
+    return "Chatbot running"
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)