Update app.py

app.py

@@ -309,4 +309,24 @@ with gr.Blocks() as app:
                 aiornot2, [inp], [outp2, n_out2]).then(
                 tot_prob, None, fin, show_progress=False)
 
+        # Tab for batch processing
+        with gr.Tab("Batch Image Processing"):
+            zip_file = gr.File(label="Upload Zip (two folders: real, ai)")
+            batch_btn = gr.Button("Process Batch")
+
+            for i, model in enumerate(models):
+                with gr.Group():
+                    gr.Markdown(f"### Results for {model}")
+                    globals()[f'output_acc{i}'] = gr.Label(label=f"Model {i} Accuracy")
+                    globals()[f'output_roc{i}'] = gr.Label(label=f"Model {i} ROC Score")
+                    globals()[f'output_report{i}'] = gr.Textbox(label=f"Model {i} Classification Report", lines=10)
+                    globals()[f'output_cm{i}'] = gr.Plot(label=f"Model {i} Confusion Matrix")
+                    globals()[f'output_roc_plot{i}'] = gr.Plot(label=f"Model {i} ROC Curve")
+
+    # Connect batch processing
+    batch_btn.click(process_zip, zip_file, 
+                    [output_acc0, output_roc0, output_report0, output_cm0, output_roc_plot0,
+                     output_acc1, output_roc1, output_report1, output_cm1, output_roc_plot1,
+                     output_acc2, output_roc2, output_report2, output_cm2, output_roc_plot2])
+
 app.launch(show_api=False, max_threads=24)