Update app.py

app.py

@@ -189,33 +189,6 @@ def get_dashboard():
     plt.tight_layout()
     return fig
 
-# ---------------- FEEDBACK ----------------
-
-def submit_feedback(result_text, feedback_choice):
-    if not result_text:
-        return "No result available."
-
-    cursor.execute(
-        "INSERT INTO feedback VALUES (?, ?)",
-        (result_text, feedback_choice)
-    )
-    conn.commit()
-    return "Feedback recorded!"
-
-def show_accuracy():
-    df = pd.read_sql("SELECT * FROM feedback", conn)
-
-    if df.empty:
-        return "No feedback yet."
-
-    correct = len(df[df["feedback"] == "Correct"])
-    total = len(df)
-
-    accuracy = (correct / total) * 100
-    return f"Accuracy (User Feedback Based): {accuracy:.2f}%"
-
-# ---------------- CALLBACK ----------------
-
 def detect_image(img, threshold):
     if img is None:
         return None, "No image provided"
@@ -235,11 +208,12 @@ with gr.Blocks() as demo:
 
     result_box = gr.Textbox(label="Detection Result", lines=4)
 
-    detect_btn = gr.Button("Detect")
+    dashboard_plot = gr.Plot()
     detect_btn.click(
-        detect_image,
-        inputs=[img_input, slider],
-        outputs=[img_output, result_box]
+    detect_image,
+    inputs=[img_input, slider],
+    outputs=[img_output, result_box, dashboard_plot]
+    )
     )
 
     gr.Markdown("### Feedback")
@@ -262,7 +236,31 @@ with gr.Blocks() as demo:
     gr.Markdown("### 📊 Dashboard")
     dashboard_plot = gr.Plot()
 
-    refresh_btn = gr.Button("Refresh Dashboard")
-    refresh_btn.click(get_dashboard, outputs=dashboard_plot)
+# ---------------- FEEDBACK ----------------
+
+def submit_feedback(result_text, feedback_choice):
+    if not result_text:
+        return "No result available."
+
+    cursor.execute(
+        "INSERT INTO feedback VALUES (?, ?)",
+        (result_text, feedback_choice)
+    )
+    conn.commit()
+    return "Feedback recorded!"
+
+def show_accuracy():
+    df = pd.read_sql("SELECT * FROM feedback", conn)
+
+    if df.empty:
+        return "No feedback yet."
+
+    correct = len(df[df["feedback"] == "Correct"])
+    total = len(df)
+
+    accuracy = (correct / total) * 100
+    return f"Accuracy (User Feedback Based): {accuracy:.2f}%"
+
+# ---------------- CALLBACK ----------------
 
 demo.launch()