Update app.py

app.py

@@ -54,4 +54,22 @@ with gr.Blocks() as demo:
 
         fig, ax = plt.subplots(figsize=(6, 3))
         ax.plot(df["Timestamp"], df["Risk_Num"], marker="o", linestyle="-", color="red")
-        a
+        ax.set_ylim(0.5, 3.5)
+        ax.set_yticks([1, 2, 3])
+        ax.set_yticklabels(['Low', 'Moderate', 'High'])
+        ax.set_title("Risk Level Over Time")
+        ax.set_xlabel("Timestamp")
+        ax.set_ylabel("Risk Level")
+        ax.tick_params(axis='x', rotation=45)
+        plt.tight_layout()
+
+        df_display = df[["Temperature", "Duration", "Risk", "Timestamp"]]
+
+        return risk, alert_msg, timestamp, summary_md, df_display, fig
+
+    predict_btn.click(classify, inputs=[temp, duration],
+                      outputs=[result, alert, ist_time, summary, history_table, plot])
+
+    retrain_btn.click(retrain_model, outputs=[retrain_output])
+
+demo.launch()