Update app.py

app.py

@@ -8,7 +8,7 @@ with gr.Blocks() as demo:
 
     # 🔄 Reset button at the top
     with gr.Row():
-        reset_btn = gr.Button("🔄 Reset All Fields")
+        reset_btn = gr.Button("🔄 Reset")
 
     # Input fields
     with gr.Row():
@@ -23,15 +23,17 @@ with gr.Blocks() as demo:
     result = gr.Textbox(label="Risk Prediction")
     alert = gr.Textbox(label="🚨 Alert Message")
     ist_time = gr.Textbox(label="Timestamp (IST)")
+    risk_score = gr.Textbox(label="📊 Risk Score")  # NEW FIELD
     summary = gr.Markdown()
     history_table = gr.Dataframe(headers=["Temperature", "Duration", "Risk", "Timestamp"], label="📈 Prediction History")
     plot = gr.Plot(label="📊 Risk Trend Chart")
 
     def classify(temp, duration):
         if temp <= 0 or duration <= 0:
-            return "❌ Invalid", "Invalid", "Invalid", "", pd.DataFrame(), plt.figure()
+            return "❌ Invalid", "Invalid", "Invalid", "", "", pd.DataFrame(), plt.figure()
 
         risk, timestamp = predict_risk(temp, duration)
+        score = round((temp + duration) / 2, 2)
 
         alert_msg = {
             "Low": "✅ SAFE - No action needed",
@@ -43,6 +45,7 @@ with gr.Blocks() as demo:
 ### 🔎 Summary
 - **Max Temp**: {temp} °C  
 - **Duration**: {duration} min  
+- **Risk Score**: {score}  
 - **Risk**: {risk}  
 - **Timestamp**: {timestamp}  
 - **Alert**: {alert_msg}
@@ -65,24 +68,24 @@ with gr.Blocks() as demo:
 
         df_display = df[["Temperature", "Duration", "Risk", "Timestamp"]]
 
-        return risk, alert_msg, timestamp, summary_md, df_display, fig
+        return risk, alert_msg, timestamp, score, summary_md, df_display, fig
 
     # Function to reset all fields
     def reset_all():
         return (
-            100, 30, "", "", "", "", pd.DataFrame(columns=["Temperature", "Duration", "Risk", "Timestamp"]), plt.figure()
+            100, 30, "", "", "", "", "", pd.DataFrame(columns=["Temperature", "Duration", "Risk", "Timestamp"]), plt.figure()
         )
 
     predict_btn.click(
         classify,
         inputs=[temp, duration],
-        outputs=[result, alert, ist_time, summary, history_table, plot]
+        outputs=[result, alert, ist_time, risk_score, summary, history_table, plot]
     )
 
     reset_btn.click(
         reset_all,
         inputs=[],
-        outputs=[temp, duration, result, alert, ist_time, summary, history_table, plot]
+        outputs=[temp, duration, result, alert, ist_time, risk_score, summary, history_table, plot]
     )
 
 demo.launch()