feat(UX): add image augmentation option for prediction

- add "augment" parameter to functions-* predict_image_with_html
- add "augment_image" function
- add extra Gradio Accordion checkbox for image augmentation

app.py

@@ -200,12 +200,15 @@ def generate_results_html(results):
     """
     return html_content
 
-def predict_image_with_html(img, confidence_threshold):
-    img_pil, results = predict_image(img, confidence_threshold)
+def predict_image_with_html(img, confidence_threshold, augment):
+    if augment:
+        img_pil, _ = augment_image(img)
+    else:
+        img_pil = img
+    img_pil, results = predict_image(img_pil, confidence_threshold)
     html_content = generate_results_html(results)
     return img_pil, html_content
-
-# Define the Gradio interface
+ 
 with gr.Blocks() as iface:
     gr.Markdown("# AI Generated Image / Deepfake Detection Models Evaluation")
     
@@ -214,7 +217,8 @@ with gr.Blocks() as iface:
             image_input = gr.Image(label="Upload Image to Analyze", sources=['upload'], type='pil')
             with gr.Accordion("Settings", open=False, elem_id="settings_accordion"):
                 confidence_slider = gr.Slider(0.0, 1.0, value=0.5, step=0.01, label="Confidence Threshold")
-            inputs = [image_input, confidence_slider]
+                augment_checkbox = gr.Checkbox(label="Augment Image", value=False)
+            inputs = [image_input, confidence_slider, augment_checkbox]
             predict_button = gr.Button("Predict")
         with gr.Column(scale=2):
             with gr.Accordion("Project OpenSight - Model Evaluations & Playground", open=False, elem_id="project_accordion"):
@@ -224,9 +228,6 @@ with gr.Blocks() as iface:
             results_html = gr.HTML(label="Model Predictions")
             outputs = [image_output, results_html]
     
-    # gr.Button("Predict").click(fn=predict_image_with_html, inputs=inputs, outputs=outputs)
-    
-    
     predict_button.click(
         fn=predict_image_with_html, 
         inputs=inputs,