Update app.py

app.py

@@ -1,6 +1,6 @@
-# backend/app.py
+# pulmoprobe_backend/app.py
 
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import joblib
@@ -129,9 +129,9 @@ def predict(data: OneHotPatientData):
             'country_Luxembourg','country_Malta','country_Netherlands','country_Poland',
             'country_Portugal','country_Romania','country_Slovakia','country_Slovenia',
             'country_Spain','country_Sweden',
-            'cancer_stage_Stage_II','cancer_stage_Stage_III','cancer_stage_Stage_IV',
+            'cancer_stage_Stage II','cancer_stage_Stage III','cancer_stage_Stage IV', # Corrected names
             'family_history_Yes',
-            'smoking_status_Former_Smoker','smoking_status_Never_Smoker','smoking_status_Passive_Smoker',
+            'smoking_status_Former Smoker','smoking_status_Never Smoked','smoking_status_Passive Smoker', # Corrected names
             'treatment_type_Combined','treatment_type_Radiation','treatment_type_Surgery'
         ]
 
@@ -141,7 +141,7 @@ def predict(data: OneHotPatientData):
 
         # Predict probabilities
         probabilities = model.predict_proba(input_df)[0]
-        confidence_high_risk = probabilities[1]  # Class 1 = High Risk
+        confidence_high_risk = probabilities[1]
         risk_level = "High Risk of Non-Survival" if confidence_high_risk > 0.5 else "Low Risk of Non-Survival"
 
         result = {
@@ -152,4 +152,4 @@ def predict(data: OneHotPatientData):
 
     except Exception as e:
         logger.error(f"Prediction error: {str(e)}")
-        return {"error": str(e), "input_data_received": data.dict()}
+        raise HTTPException(status_code=500, detail=str(e))