Update app.py

app.py

@@ -364,6 +364,8 @@ def create_gradio_interface():
                     output_plots = gr.Plot(label="Confusion Matrix and ROC Curve")
                     output_fp_fn = gr.HTML(label="False Positives and Negatives")
 
+                reset_btn = gr.Button("Reset")
+
         load_btn.click(load_url, in_url, [inp, message])
         btn.click(
             lambda img: detector.predict(img),
@@ -374,4 +376,50 @@ def create_gradio_interface():
         def enable_zip_btn(file):
             return gr.Button.update(interactive=file is not None)
 
-        def enable_individual_btn
+        def enable_individual_btn(ai_files, real_files):
+            return gr.Button.update(interactive=(ai_files is not None and real_files is not None))
+
+        zip_file.upload(enable_zip_btn, zip_file, zip_process_btn)
+        
+        ai_files.upload(enable_individual_btn, [ai_files, real_files], individual_process_btn)
+        real_files.upload(enable_individual_btn, [ai_files, real_files], individual_process_btn)
+
+        zip_process_btn.click(
+            process_zip,
+            zip_file,
+            [output_acc, output_roc, output_report, output_plots, output_fp_fn]
+        )
+
+        individual_process_btn.click(
+            process_files,
+            [ai_files, real_files],
+            [output_acc, output_roc, output_report, output_plots, output_fp_fn]
+        )
+
+        def reset_interface():
+            return [
+                None, None, None, None, None,  # Reset inputs
+                gr.Button.update(interactive=False),  # Reset zip process button
+                gr.Button.update(interactive=False),  # Reset individual process button
+                None, None, None, None, None  # Reset outputs
+            ]
+
+        reset_btn.click(
+            reset_interface,
+            inputs=None,
+            outputs=[
+                zip_file, ai_files, real_files,
+                output_acc, output_roc, output_report, output_plots, output_fp_fn,
+                zip_process_btn, individual_process_btn
+            ]
+        )
+
+    return app
+
+if __name__ == "__main__":
+    app = create_gradio_interface()
+    app.launch(
+        show_api=False,
+        max_threads=24,
+        show_error=True
+    )