Update app.py

app.py

@@ -1,44 +1,170 @@
+# import gradio as gr
+# import subprocess
+# import os
+# import random
+# from PIL import Image
+
+# # Paths
+# OUTPUT_DIR = "outputs"
+# MODEL_PATH = "blazer_model.pkl"  # Adjusted for local Hugging Face repo structure
+
+# # Ensure the output directory exists
+# os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+# # Function to generate images using StyleGAN3
+# def generate_images():
+#     command = f"python stylegan3/gen_images.py --outdir={OUTPUT_DIR} --trunc=1 --seeds='3-5,7,9,12-14,16-26,29,31,32,34,40,41' --network={MODEL_PATH}"
+#     try:
+#         subprocess.run(command, shell=True, check=True)
+#     except subprocess.CalledProcessError as e:
+#         return f"Error generating images: {e}"
+
+# # Function to select 5 random images
+# 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")]
+#     random_images = random.sample(image_files, min(10, len(image_files)))
+#     return [Image.open(os.path.join(OUTPUT_DIR, img)) for img in random_images]
+
+# # Gradio function
+# def generate_and_display():
+#     generate_images()
+#     return get_random_images()
+
+# # UI
+# with gr.Blocks() as demo:
+#     gr.Markdown("# 🎨 AI-Generated Clothing Designs - Blazers")
+#     generate_button = gr.Button("Generate New Designs")
+#     output_gallery = gr.Gallery(label="Generated Designs", columns=5, rows=2)
+#     generate_button.click(fn=generate_and_display, outputs=output_gallery)
+
+# if __name__ == "__main__":
+#     demo.launch()
+
+
+
+
 import gradio as gr
 import subprocess
 import os
 import random
 from PIL import Image
+import shutil
+import requests
 
-# Paths
-OUTPUT_DIR = "outputs"
-MODEL_PATH = "blazer_model.pkl"  # Adjusted for local Hugging Face repo structure
+# === 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, "blazer_model.pkl")
+SAVE_DIR = os.path.join(BASE_DIR, "saved_images")
 
-# Ensure the output directory exists
 os.makedirs(OUTPUT_DIR, exist_ok=True)
+os.makedirs(SAVE_DIR, exist_ok=True)
 
-# Function to generate images using StyleGAN3
+# === Image Generation Function ===
 def generate_images():
-    command = f"python stylegan3/gen_images.py --outdir={OUTPUT_DIR} --trunc=1 --seeds='3-5,7,9,12-14,16-26,29,31,32,34,40,41' --network={MODEL_PATH}"
+    command = [
+        "python",
+        GEN_SCRIPT,
+        f"--outdir={OUTPUT_DIR}",
+        "--trunc=1",
+        "--seeds=3-5,7,9,12-14,16-26,29,31,32,34,40,41",
+        f"--network={MODEL_PATH}"
+    ]
     try:
-        subprocess.run(command, shell=True, check=True)
+        subprocess.run(command, check=True, capture_output=True, text=True)
     except subprocess.CalledProcessError as e:
-        return f"Error generating images: {e}"
+        return f"Error generating images:\n{e.stderr}"
 
-# Function to select 5 random images
+# === 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")]
     random_images = random.sample(image_files, min(10, len(image_files)))
-    return [Image.open(os.path.join(OUTPUT_DIR, img)) for img in random_images]
+    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."
+
+    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')}
+
+            # Your backend endpoint here
+            url = f" https://a56e-103-40-74-78.ngrok-free.app/images/upload/{user_id}"
+            response = requests.post(url, files=files)
 
-# Gradio function
-def generate_and_display():
-    generate_images()
-    return get_random_images()
+        if response.status_code == 201:
+            return "✅ Image uploaded and saved to database!"
+        else:
+            return f"❌ Upload failed: {response.status_code} - {response.text}"
 
-# UI
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}"
+
+# === Gradio Interface ===
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎨 AI-Generated Clothing Designs - Blazers")
+    gr.Markdown("# 🎨 AI-Generated Clothing Designs - Dresses")
+
     generate_button = gr.Button("Generate New Designs")
-    output_gallery = gr.Gallery(label="Generated Designs", columns=5, rows=2)
-    generate_button.click(fn=generate_and_display, outputs=output_gallery)
+    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 = []
+
+    # Use 3 columns layout
+    for row_idx in range(4):  # 4 rows (to cover 10 images)
+        with gr.Row():
+            for col_idx in range(3):  # 3 columns
+                i = row_idx * 3 + col_idx
+                if i >= 10:
+                    break
+                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)
+
+                    save_btn.click(
+                        fn=send_to_backend,
+                        inputs=[file_path, user_id_state],
+                        outputs=output
+                    )
+
+    # Generate button logic
+    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
+    )
 
 if __name__ == "__main__":
     demo.launch()
+