Update app.py

app.py

@@ -1,34 +1,77 @@
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import uvicorn
 from medical_simplifier import MedicalTextSimplifier
+import logging
 
-# Initialize Flask app
-app = Flask(__name__)
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# Initialize your simplifier once at startup
-simplifier = MedicalTextSimplifier()
+app = FastAPI(
+    title="Medical Text Simplifier API",
+    description="API to simplify medical text by explaining medical terms in plain language",
+    version="1.0.0"
+)
 
-@app.route('/simplify', methods=['POST'])
-def simplify():
-    data = request.get_json()
-    if not data or 'text' not in data:
-        return jsonify({'error': 'Missing "text" in request body'}), 400
+# Initialize the simplifier (this will load the models)
+try:
+    simplifier = MedicalTextSimplifier()
+    logger.info("Medical text simplifier initialized successfully")
+except Exception as e:
+    logger.error(f"Failed to initialize simplifier: {e}")
+    simplifier = None
 
-    text = data['text']
+class TextInput(BaseModel):
+    text: str
+
+class SimplificationResponse(BaseModel):
+    original_text: str
+    medical_terms_explained: dict
+
+@app.get("/")
+async def root():
+    return {
+        "message": "Medical Text Simplifier API",
+        "status": "running",
+        "endpoints": ["/simplify", "/health"]
+    }
+
+@app.get("/health")
+async def health_check():
+    if simplifier is None:
+        raise HTTPException(status_code=503, detail="Model not loaded")
+    return {"status": "healthy", "models_loaded": True}
+
+@app.post("/simplify", response_model=SimplificationResponse)
+async def simplify_text(input_data: TextInput):
+    if simplifier is None:
+        raise HTTPException(status_code=503, detail="Model not loaded")
+    
+    if not input_data.text.strip():
+        raise HTTPException(status_code=400, detail="Text input cannot be empty")
+    
     try:
-        simplified = simplifier.simplify_text(text)
-        return jsonify({
-            'original_text': text,
-            'simplified_text': simplified
-        })
+        # Get medical terms
+        medical_terms = simplifier.identify_medical_terms(input_data.text)
+        
+        # Create dictionary to store unique terms and their explanations
+        unique_terms = {}
+        
+        for item in medical_terms:
+            term = item['term'].lower()  # Convert to lowercase for comparison
+            if term not in unique_terms:
+                explanation = simplifier.generate_simplified_explanation(item['term'], input_data.text)
+                unique_terms[term] = explanation
+        
+        return SimplificationResponse(
+            original_text=input_data.text,
+            medical_terms_explained=unique_terms
+        )
+    
     except Exception as e:
-        return jsonify({'error': str(e)}), 500
-
-@app.route('/', methods=['GET'])
-def home():
-    return jsonify({
-        'message': 'Medical Text Simplifier API is running.',
-        'usage': 'POST text to /simplify to get simplified medical explanation.'
-    })
+        logger.error(f"Error processing text: {e}")
+        raise HTTPException(status_code=500, detail="Error processing text")
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=50010)
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)