Update app.py

app.py

@@ -69,7 +69,6 @@
 
 
 
-
 # import gradio as gr
 # import subprocess
 # import os
@@ -77,14 +76,17 @@
 # from PIL import Image
 # import shutil
 
+# # === Setup Paths ===
 # BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 # GEN_SCRIPT = os.path.join(BASE_DIR, "stylegan3", "gen_images.py")
 # OUTPUT_DIR = os.path.join(BASE_DIR, "outputs")
 # MODEL_PATH = os.path.join(BASE_DIR, "dress.pkl")
+# SAVE_DIR = os.path.join(BASE_DIR, "saved_images")
 
 # os.makedirs(OUTPUT_DIR, exist_ok=True)
-# os.makedirs(os.path.join(BASE_DIR, "saved_images"), exist_ok=True)
+# os.makedirs(SAVE_DIR, exist_ok=True)
 
+# # === Image Generation Function ===
 # def generate_images():
 #     command = [
 #         "python",
@@ -99,62 +101,64 @@
 #     except subprocess.CalledProcessError as e:
 #         return f"Error generating images:\n{e.stderr}"
 
+# # === Select Random Images from Output Folder ===
 # def get_random_images():
 #     image_files = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(".png")]
 #     if len(image_files) < 10:
 #         generate_images()
 #         image_files = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(".png")]
-#     selected_files = random.sample(image_files, min(10, len(image_files)))
-#     return [os.path.join(OUTPUT_DIR, f) for f in selected_files]
+#     random_images = random.sample(image_files, min(10, len(image_files)))
+#     image_paths = [os.path.join(OUTPUT_DIR, img) for img in random_images]
+#     return image_paths
 
-# def save_image(image_path):
-#     filename = os.path.basename(image_path)
-#     dest = os.path.join(BASE_DIR, "saved_images", filename)
-#     shutil.copy(image_path, dest)
-#     return f"✅ Saved {filename}"
+# # === Save Image Function ===
+# def save_image(img_path):
+#     try:
+#         filename = os.path.basename(img_path)
+#         dest_path = os.path.join(SAVE_DIR, filename)
+#         shutil.copy(img_path, dest_path)
+#         return f"Saved to: {dest_path}"
+#     except Exception as e:
+#         return f"Error saving image: {str(e)}"
 
+# # === Gradio Interface ===
 # with gr.Blocks() as demo:
-#     gr.Markdown("## 👗 AI-Generated Clothing Designs")
+#     gr.Markdown("# 🎨 AI-Generated Clothing Designs - Dresses")
 
-#     generate_btn = gr.Button("✨ Generate New Designs")
+#     generate_button = gr.Button("Generate New Designs")
 
 #     image_components = []
-#     file_paths = []
 #     save_buttons = []
 #     outputs = []
+#     file_paths = []
 
 #     with gr.Row():
-#         for row in range(2):  # 2 rows
-#             with gr.Column():
-#                 for col in range(5):  # 5 images per row
-#                     index = row * 5 + col
-#                     with gr.Column():
-#                         img = gr.Image()
-#                         file_paths.append(gr.Textbox(visible=False))  # to pass file path
-#                         btn = gr.Button("💾 Save")
-#                         out = gr.Textbox(visible=False)
+#         for i in range(10):  # 10 images
+#             with gr.Column(scale=1):
+#                 img = gr.Image(width=200, height=200)
+#                 image_components.append(img)
 
-#                         btn.click(fn=save_image, inputs=file_paths[index], outputs=out)
+#                 file_path = gr.Textbox(visible=False)
+#                 file_paths.append(file_path)
 
-#                         image_components.append(img)
-#                         save_buttons.append(btn)
-#                         outputs.append(out)
+#                 save_btn = gr.Button("💾 Save")
+#                 output = gr.Textbox(visible=False)
+#                 outputs.append(output)
 
-#     def update_gallery():
-#         files = get_random_images()
-#         updates = []
-#         for i, path in enumerate(files):
-#             updates.append(path)  # for img.update(value=...)
-#             updates.append(path)  # for file_path.update(value=...)
-#         # Fill remaining slots if less than 10
-#         for i in range(10 - len(files)):
-#             updates.append(None)
-#             updates.append("")
-#         return updates
+#                 save_btn.click(fn=save_image, inputs=file_path, outputs=output)
+#                 save_buttons.append(save_btn)
 
-#     generate_btn.click(fn=update_gallery, outputs=sum(zip(image_components, file_paths), ()))
+#     def generate_and_display_images():
+#         image_paths = get_random_images()
+#         return image_paths + image_paths  # one set for images, one for hidden file paths
 
-# demo.launch()
+#     generate_button.click(
+#         fn=generate_and_display_images,
+#         outputs=image_components + file_paths
+#     )
+
+# if __name__ == "__main__":
+#     demo.launch()
 
 
 
@@ -163,17 +167,19 @@ import subprocess
 import os
 import random
 from PIL import Image
-import shutil
+import requests
+import io
+import warnings
+
+warnings.filterwarnings("ignore")
 
 # === Setup Paths ===
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 GEN_SCRIPT = os.path.join(BASE_DIR, "stylegan3", "gen_images.py")
 OUTPUT_DIR = os.path.join(BASE_DIR, "outputs")
 MODEL_PATH = os.path.join(BASE_DIR, "dress.pkl")
-SAVE_DIR = os.path.join(BASE_DIR, "saved_images")
 
 os.makedirs(OUTPUT_DIR, exist_ok=True)
-os.makedirs(SAVE_DIR, exist_ok=True)
 
 # === Image Generation Function ===
 def generate_images():
@@ -200,18 +206,39 @@ def get_random_images():
     image_paths = [os.path.join(OUTPUT_DIR, img) for img in random_images]
     return image_paths
 
-# === Save Image Function ===
-def save_image(img_path):
+# === Upload to NestJS Backend ===
+def send_to_backend(img_path, user_id):
+    if not user_id:
+        return "❌ user_id not found."
+    if not os.path.exists(img_path):
+        return "⚠️ Image path invalid."
+
     try:
-        filename = os.path.basename(img_path)
-        dest_path = os.path.join(SAVE_DIR, filename)
-        shutil.copy(img_path, dest_path)
-        return f"Saved to: {dest_path}"
+        with open(img_path, "rb") as f:
+            file_bytes = f.read()
+
+        files = {'file': ('generated_image.png', file_bytes, 'image/png')}
+
+        url = f"https://5a4d-103-40-74-78.ngrok-free.app/customisation/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}"
+
     except Exception as e:
-        return f"Error saving image: {str(e)}"
+        return f"⚠️ Error: {str(e)}"
 
 # === Gradio Interface ===
 with gr.Blocks() as demo:
+    user_id_state = gr.State()
+
+    @demo.load(inputs=None, outputs=[user_id_state])
+    def on_load(request: gr.Request):
+        user_id = request.query_params.get('user_id', '')
+        return user_id
+
     gr.Markdown("# 🎨 AI-Generated Clothing Designs - Dresses")
 
     generate_button = gr.Button("Generate New Designs")
@@ -234,12 +261,17 @@ with gr.Blocks() as demo:
                 output = gr.Textbox(visible=False)
                 outputs.append(output)
 
-                save_btn.click(fn=save_image, inputs=file_path, outputs=output)
+                # Save button uses the backend-saving version
+                save_btn.click(
+                    fn=send_to_backend,
+                    inputs=[file_path, user_id_state],
+                    outputs=output
+                )
                 save_buttons.append(save_btn)
 
     def generate_and_display_images():
         image_paths = get_random_images()
-        return image_paths + image_paths  # one set for images, one for hidden file paths
+        return image_paths + image_paths  # one set for images, one for file paths
 
     generate_button.click(
         fn=generate_and_display_images,