Update app.py

app.py

@@ -5,7 +5,7 @@ from risk_model import predict_risk, retrain_model, get_history_df
 # Create FastAPI instance
 fastapi_app = FastAPI()
 
-# Create Gradio app
+# Gradio UI blocks
 gradio_app = gr.Blocks()
 
 with gradio_app:
@@ -23,7 +23,10 @@ with gradio_app:
     score = gr.Textbox(label="Confidence (%)")
     retrain_output = gr.Textbox(label="Retrain Status")
 
-    history_table = gr.Dataframe(headers=["Temperature", "Duration", "Risk", "Confidence"], label="📈 Prediction History")
+    history_table = gr.Dataframe(
+        headers=["Temperature", "Duration", "Risk", "Confidence"],
+        label="📈 Prediction History"
+    )
 
     def classify(temp, duration):
         if temp <= 0 or duration <= 0:
@@ -35,10 +38,10 @@ with gradio_app:
     predict_btn.click(classify, inputs=[temp, duration], outputs=[result, score, history_table])
     retrain_btn.click(retrain_model, outputs=[retrain_output])
 
-# Optional: basic route to test API is up
+# Optional test route
 @fastapi_app.get("/")
 def root():
-    return {"status": "Heating Mantle API is live"}
+    return {"message": "Heating Mantle FastAPI is working"}
 
-# ✅ Hugging Face requires this line exactly
+# ✅ Hugging Face requires this:
 app = gr.mount_gradio_app(fastapi_app, gradio_app, path="/predict-ui")