Spaces:

capagio
/

garment_designer

Paused

App Files Files Community

giorgio-caparvi commited on Aug 25, 2024

Commit

179259d

1 Parent(s): 5457fbb

Button to download the image

Browse files

Files changed (3) hide show

api/app.py +8 -29
api/templates/index.html +19 -25
api/templates/output.html +12 -0

api/app.py CHANGED Viewed

@@ -1,16 +1,10 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory
-from flask_cors import CORS
 import subprocess
 import os
 app = Flask(__name__)
-CORS(app)
-# Ensure that Flask looks for the HTML templates in the correct folder
-app.template_folder = './templates'
-print(f"Current working directory: {os.getcwd()}")
-# Set the directory for the output images
 output_dir = '/api/output/generato_paired_paired/images'
 image_filename = '03191_00.jpg'
@@ -20,7 +14,7 @@ def index():
 @app.route('/generate-design', methods=['POST'])
 def generate_design():
-    # Command handling
     command = [
         'python', '/api/model/src/eval.py',
         '--dataset_path', '/api/model/assets/data/vitonhd',
@@ -35,27 +29,12 @@ def generate_design():
         '--test_order', 'paired'
     ]
-    try:
-        result = subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        print(f"Command output: {result.stdout}")
-        print(f"Command errors: {result.stderr}")
-        #image_path = os.path.join(output_dir, image_filename)
-        image_path = output_dir#"/api/output/generato_paired_paired/images/03191_00.jpg"
-        print("immagine creata")
-        print(f"Directory Exists: {os.path.exists(output_dir)}")
-        print(f"Can Write: {os.access(output_dir, os.W_OK)}")
-        # Check if the image was generated
-        if os.path.exists(image_path):
-            print("IMMAGINE CREATA")
-            return jsonify({"status": "success", "image_url": f"{output_dir}"})
-        else:
-            print("IMMAGINE NON CREATA")
-            return jsonify({"status": "error", "message": "Image generation failed"}), 500
-    except subprocess.CalledProcessError as e:
-        return jsonify({"status": "error", "message": str(e)}), 500
-# Route to serve the generated image
 @app.route('/api/output/generato_paired_paired/images/<filename>')
 def serve_image(filename):
     return send_from_directory(output_dir, filename)

+from flask import Flask, render_template, send_from_directory
 import subprocess
 import os
 app = Flask(__name__)
+# Directory di output
 output_dir = '/api/output/generato_paired_paired/images'
 image_filename = '03191_00.jpg'
 @app.route('/generate-design', methods=['POST'])
 def generate_design():
+    # Esegui il comando di generazione
     command = [
         'python', '/api/model/src/eval.py',
         '--dataset_path', '/api/model/assets/data/vitonhd',
         '--test_order', 'paired'
     ]
+    # Esegui il comando senza gestire il JSON, lascia che l'errore venga stampato
+    subprocess.run(command, check=True)
+    return '', 204  # Nessun contenuto
+# Route per servire l'immagine generata
 @app.route('/api/output/generato_paired_paired/images/<filename>')
 def serve_image(filename):
     return send_from_directory(output_dir, filename)

api/templates/index.html CHANGED Viewed

@@ -3,48 +3,42 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Generate Design</title>
     <script>
         function generateDesign() {
-            // Remove the previous image, if any
-            document.getElementById("generated-image").src = "";
-            // Disable the button to prevent multiple requests
             const button = document.getElementById("generate-button");
             button.disabled = true;
             button.innerText = "Processing...";
-            fetch('/generate-design', {
-                method: 'POST'
-            })
-            .then(response => response.json())
-            .then(data => {
-                if (data.status === "success") {
-                    // Display the generated image
-                    const imgElement = document.getElementById("generated-image");
-                    imgElement.src = data.image_url;
-                    imgElement.style.display = "block";
-                } else {
-                    alert("Error: " + data.message);
-                }
             })
             .catch(error => {
                 console.error("Error:", error);
-                alert("Failed to execute the command.");
-            })
-            .finally(() => {
-                // Re-enable the button
                 button.disabled = false;
                 button.innerText = "Run Command";
             });
         }
     </script>
 </head>
 <body>
     <h1>Generate Design</h1>
-    <button id="generate-button" onclick="generateDesign()">Run Command</button>
-    <!-- Container for displaying the generated image -->
-    <img id="generated-image" style="display:none; margin-top: 20px;" alt="Generated Design" />
 </body>
 </html>

 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Generate and Download Design</title>
     <script>
         function generateDesign() {
+            // Disabilita il pulsante per evitare richieste multiple
             const button = document.getElementById("generate-button");
             button.disabled = true;
             button.innerText = "Processing...";
+            // Esegui la chiamata fetch per attivare il comando
+            fetch('/generate-design', { method: 'POST' })
+            .then(() => {
+                // Abilita il pulsante di download
+                document.getElementById("download-button").style.display = "block";
+                button.disabled = false;
+                button.innerText = "Run Command";
             })
             .catch(error => {
                 console.error("Error:", error);
                 button.disabled = false;
                 button.innerText = "Run Command";
+                alert("Failed to execute the command.");
             });
         }
     </script>
 </head>
 <body>
     <h1>Generate Design</h1>
+    <!-- Bottone per attivare la generazione -->
+    <button id="generate-button" onclick="generateDesign()">Run Command</button>
+    <br><br>
+    <!-- Bottone per il download (nascosto fino a che non è pronto) -->
+    <a id="download-button" href="/api/output/generato_paired_paired/images/03191_00.jpg" download style="display:none;">
+        Download Image
+    </a>
 </body>
 </html>

api/templates/output.html ADDED Viewed

	@@ -0,0 +1,12 @@

+<!DOCTYPE html>
+<html>
+<head>
+    <title>Image Generated</title>
+</head>
+<body>
+    <h1>Design Generated Successfully</h1>
+    <img src="{{ image_url }}" alt="Generated Image" style="max-width: 100%; height: auto;">
+    <br>
+    <a href="{{ download_url }}" download>Download Image</a>
+</body>
+</html>