add save button

app.py

@@ -3,6 +3,7 @@ import cvzone
 import numpy as np
 import os
 import gradio as gr
+from datetime import datetime
 
 # Load the YuNet model
 model_path = 'face_detection_yunet_2023mar.onnx'
@@ -118,6 +119,31 @@ def transform_cv2(frame, transform):
     else:
         return frame
     
+def refresh_interface():
+    # # Reset the transformation dropdown to its default value
+    # transform.update(value="none")
+
+    # Reset the image to an empty state or a default image
+    input_img.update(value=None)
+
+    # Return a message indicating the interface has been refreshed
+    return "Interface refreshed!"
+
+def save_frame(frame):
+       # Convert frame to RGB
+       frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+
+       # Create a unique filename using the current timestamp
+       filename = f"saved_frame_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
+
+       # Save the frame
+       cv2.imwrite(filename, frame)
+
+       # Refresh the interfaceq
+       refresh_interface()
+
+       return f"Frame saved as '{filename}'"
+
 
 
 # Gradio webcam input
@@ -134,21 +160,14 @@ with gr.Blocks() as demo:
                                     value="none", label="Transformation")
             input_img = gr.Image(sources=["webcam"], type="numpy", streaming=True)
             next_button = gr.Button("Next Glasses")
+            save_button = gr.Button("Save as a Picture") 
     input_img.stream(webcam_input, [input_img, transform], [input_img], time_limit=30, stream_every=0.1)
-
-    next_button.click(change_glasses, [], [])
+    with gr.Row():
+        next_button.click(change_glasses, [], [])
+    with gr.Row():
+        save_button.click(save_frame, [input_img], [])
 if __name__ == "__main__":
     demo.launch(share=True)
 
 
 
-# # Gradio Interface
-# with gr.Blocks() as demo:
-#     with gr.Row():
-#         with gr.Column():
-#             input_img = gr.Image(label="Input", sources="webcam", streaming=True)
-#             next_button = gr.Button("Next Glasses")
-#     input_img.stream(webcam_input, [input_img], [input_img], stream_every=0.1, concurrency_limit=30)
-#     next_button.click(change_glasses, [], [])
-# if __name__ == "__main__":
-#     demo.launch(share=True)