Update app.py

app.py

@@ -6,19 +6,24 @@ from deep_translator import GoogleTranslator
 app = Flask(__name__)
 CORS(app)
 
-# Load disease data
+# 🔹 Load disease data
 with open("diseases.json", "r", encoding="utf-8") as f:
     disease_data = json.load(f)
 
+# 🔹 Disease aliases
 DISEASE_ALIASES = {
+    "vascular": "vascular lesion",
+    "bcc": "Basal Cell Carcinoma (BCC)",
+    "basal cell": "Basal Cell Carcinoma (BCC)",
+    "scc": "squamous cell carcinoma",
     "ringworm": "Tinea Ringworm Candidiasis",
     "fungal infection": "Tinea Ringworm Candidiasis",
-    "eczema": "eczema",
     "warts": "Warts Molluscum Viral Infections",
-    "melanoma": "melanoma",
-    "nevus": "nevus"
+    "mole": "nevus",
+    "pigmented keratosis": "pigmented benign keratosis"
 }
 
+# 🌍 Supported languages
 SUPPORTED_LANGUAGES = {
     "en": "English",
     "ta": "Tamil",
@@ -39,16 +44,13 @@ def home():
 @app.route("/chat", methods=["POST"])
 def chat():
     data = request.json
-
     user_input = data.get("message", "")
-    user_lang = data.get("language", "en")  # 🔥 THIS IS KEY
-
-    print("LANG RECEIVED 👉", user_lang)   # 🔥 DEBUG (HF logs)
+    user_lang = data.get("language", "en")  # 👈 dropdown language
 
     if user_lang not in SUPPORTED_LANGUAGES:
         user_lang = "en"
 
-    # Translate input to English
+    # 🔄 Translate input → English
     if user_lang != "en":
         translated_input = GoogleTranslator(
             source=user_lang, target="en"
@@ -58,25 +60,39 @@ def chat():
 
     normalized_input = translated_input.lower()
 
+    # 🔍 Alias match
     matched_disease = None
     for key, value in DISEASE_ALIASES.items():
         if key in normalized_input:
             matched_disease = value
             break
 
+    # 🔍 Direct match
+    if not matched_disease:
+        for disease in disease_data:
+            if disease.lower() in normalized_input:
+                matched_disease = disease
+                break
+
     if not matched_disease or matched_disease not in disease_data:
-        return jsonify({"reply": "Disease information not found"})
+        return jsonify({
+            "reply": "Disease information not found."
+        })
 
     info = disease_data[matched_disease]
 
+    # 🧾 English base response
     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"Causes:\n➤ " + "\n➤ ".join(info["causes"]) + "\n\n"
         f"Precautions:\n➤ " + "\n➤ ".join(info["precautions"]) + "\n\n"
-        "⚠ Educational purpose only"
+        f"Treatment:\n➤ " + "\n➤ ".join(info["treatment"]) + "\n\n"
+        "⚠ Educational purpose only. Consult a dermatologist."
     )
 
+    # 🌍 Translate output
     if user_lang != "en":
         final_response = GoogleTranslator(
             source="en", target=user_lang
@@ -84,7 +100,10 @@ def chat():
     else:
         final_response = english_response
 
-    return jsonify({"reply": final_response})
+    return jsonify({
+        "reply": final_response,
+        "language": SUPPORTED_LANGUAGES[user_lang]
+    })
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)