File size: 2,484 Bytes
5441c26 418a820 5441c26 418a820 5441c26 418a820 5441c26 418a820 5441c26 418a820 5441c26 418a820 5441c26 418a820 5441c26 418a820 5441c26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import uvicorn
from medical_simplifier import MedicalTextSimplifier
import logging
# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
app = FastAPI(
title="Medical Text Simplifier API",
description="API to simplify medical text by explaining medical terms in plain language",
version="1.0.0"
)
# 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
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:
# 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:
logger.error(f"Error processing text: {e}")
raise HTTPException(status_code=500, detail="Error processing text")
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=7860) |