|
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) |
