from ai import DiseasePredictionModel
from flask import Flask, request, jsonify

app = Flask(__name__)

model = DiseasePredictionModel()

@app.route("/ai/text", methods=["POST"])
def ai_text():
    try:
        text = request.json["prompt"]
        print(f"Original text: {text}")
        corrected_text = model.correct_text(text)
        print(f"Corrected text: {corrected_text}")
        extracted_symptoms = model.extract_symptoms(corrected_text)
        print(f"Extracted symptoms: {extracted_symptoms}")
        matched_symptoms = model.match_symptoms(extracted_symptoms)
        print(f"Matched symptoms: {matched_symptoms}")
        if matched_symptoms == []:
            return jsonify({"disease": "No matching symptoms found."})
        predicted_disease = model.predict_disease(matched_symptoms)
        print(f"Predicted disease: {predicted_disease}")
        return jsonify({"disease": predicted_disease})
    except:
        return 400
    
    

@app.route("/ai/list", methods=["POST"])
def ai_list():
    body = request.get_json()
    try:
        data = body["symptoms"]
        print(data)
        out = model.predict_disease(data)
        return jsonify({"output":out}),200    
    except:
        return jsonify({"status": "error"}),400
    
if __name__ == "__main__":
    app.run(debug=True, host="0.0.0.0", port=7860)