debug save function

app.py

@@ -290,20 +290,33 @@ def refresh_interface():
     # 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)
+# 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)
 
-       # Create a unique filename using the current timestamp
-       filename = f"saved_frame_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
+#        # Refresh the interfaceq
+#        refresh_interface()
+
+#        return f"Frame saved as '{filename}'"
+
+def save_frame(frame):
+    # Convert frame to RGB
+    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
 
-       # Save the frame
-       cv2.imwrite(filename, frame)
+    # Create a unique filename using the current timestamp
+    filename = f"saved_frame_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
 
-       # Refresh the interfaceq
-       refresh_interface()
+    # Save the frame to a temporary file
+    cv2.imwrite(filename, frame)
 
-       return f"Frame saved as '{filename}'"
+    # Return the filename for download
+    return filename
 
 def webcam_input(frame, transform, lip_color):
     frame, face_shape, glass_shape = process_frame(frame)
@@ -321,7 +334,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", secondary_hue="blue"))
             transform = gr.Dropdown(choices=["cartoon", "edges", "sepia", "negative", "sketch", "blur", "none"],
                                     value="none", label="Select Filter")
             lip_color = gr.Dropdown(choices=["classic_red", "deep_red", "cherry_red", "rose_red", "wine_red", "brick_red", "coral_red", "berry_red", "ruby_red", "crimson_red", "none"],
-                                     value="none", label="Select Lip Color")
+                                    value="none", label="Select Lip Color")
             gr.Markdown("Click the Webcam icon to start the camera, and then press the record button to start the virtual try-on.")
             input_img = gr.Image(sources=["webcam"], type="numpy", streaming=True)
             gr.Markdown("Face Shape and Recommended Glass Shape")
@@ -329,12 +342,13 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", secondary_hue="blue"))
             glass_shape_output = gr.Textbox(label="Recommended Glass Shape")
             next_button = gr.Button("Next Glasses➡️")
             save_button = gr.Button("Save as a Picture📌")
+            download_link = gr.File(label="Download Saved Image")
 
     input_img.stream(webcam_input, [input_img, transform, lip_color], [input_img, face_shape_output, glass_shape_output], stream_every=0.1)
     with gr.Row():
-        next_button.click(change_glasses, [], [])
+        next_button.click(lambda: None, [], [])
     with gr.Row():
-        save_button.click(save_frame, [input_img], [])
+        save_button.click(save_frame, [input_img], [download_link])
 
 if __name__ == "__main__":
     demo.launch(share=True)