Update app.py

app.py

@@ -41,29 +41,29 @@
 #     image_paths = [os.path.join(OUTPUT_DIR, img) for img in random_images]
 #     return image_paths
 
-# # === Send Image to Backend ===
-# def send_to_backend(img_path, user_id):
-#     if not user_id:
-#         return "❌ user_id not found in URL."
+# === Send Image to Backend ===
+def send_to_backend(img_path, user_id):
+    if not user_id:
+        return "❌ user_id not found in URL."
 
-#     if not img_path or not os.path.exists(img_path):
-#         return "⚠️ No image selected or image not found."
+    if not img_path or not os.path.exists(img_path):
+        return "⚠️ No image selected or image not found."
 
-#     try:
-#         with open(img_path, 'rb') as f:
-#             files = {'file': ('generated_image.png', f, 'image/png')}
+    try:
+        with open(img_path, 'rb') as f:
+            files = {'file': ('generated_image.png', f, 'image/png')}
 
-#             # Your backend endpoint here
-#             url = f" https://7da2-2409-4042-6e81-1806-de6-b8e5-836c-6b95.ngrok-free.app/images/upload/{user_id}"
-#             response = requests.post(url, files=files)
+            # Your backend endpoint here
+            url = f" https://7da2-2409-4042-6e81-1806-de6-b8e5-836c-6b95.ngrok-free.app/images/upload/{user_id}"
+            response = requests.post(url, files=files)
 
-#         if response.status_code == 201:
-#             return "✅ Image uploaded and saved to database!"
-#         else:
-#             return f"❌ Upload failed: {response.status_code} - {response.text}"
+        if response.status_code == 201:
+            return "✅ Image uploaded and saved to database!"
+        else:
+            return f"❌ Upload failed: {response.status_code} - {response.text}"
 
-#     except Exception as e:
-#         return f"⚠️ Error: {str(e)}"
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}"
 
 # # === Gradio Interface ===
 # with gr.Blocks() as demo:
@@ -107,15 +107,15 @@
 #                         outputs=output
 #                     )
 
-#     # Generate button logicg
-#     def generate_and_display_images():
-#         image_paths = get_random_images()
-#         return image_paths + image_paths  # One for display, one for hidden path tracking
+    # Generate button logicg
+    def generate_and_display_images():
+        image_paths = get_random_images()
+        return image_paths + image_paths  # One for display, one for hidden path tracking
 
-#     generate_button.click(
-#         fn=generate_and_display_images,
-#         outputs=image_components + file_paths
-#     )
+    # generate_button.click(
+    #     fn=generate_and_display_images,
+    #     outputs=image_components + file_paths
+    # )
 
 # if __name__ == "__main__":
 #     demo.launch()
@@ -189,8 +189,37 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
 
     with gr.Row():
         submit_button = gr.Button("Generate Designs")
+        user_id_state = gr.State()
+
+        @demo.load(inputs=None, outputs=[user_id_state])
+        def get_user_id(request: gr.Request):
+            return request.query_params.get("user_id", "")
+    
+        image_components = []
+        file_paths = []
+        save_buttons = []
+        outputs = []
+
+    with gr.Row():
+    for i in range(2):  # Only 2 images
+        with gr.Column():
+            img = gr.Image(width=180, height=180, label=f"Design {i+1}")
+            image_components.append(img)
+
+            file_path = gr.Textbox(visible=False)
+            file_paths.append(file_path)
+
+            save_btn = gr.Button("💾 Save to DB")
+            save_buttons.append(save_btn)
+
+            output = gr.Textbox(label="Status", interactive=False)
+            outputs.append(output)
 
-    output_gallery = gr.Gallery(label="Generated Designs", show_label=True, columns=2, height="auto")
+            save_btn.click(
+                fn=send_to_backend,
+                inputs=[file_path, user_id_state],
+                outputs=output
+            )
 
 
     examples = gr.Examples(
@@ -201,6 +230,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
         inputs=[input_box]
     )
 
-    submit_button.click(fn=generate_top_dresses, inputs=input_box, outputs=output_gallery)
+    submit_button.click(
+        fn=generate_and_display_images,
+        outputs=image_components + file_paths
 
 demo.launch()