Upload main.py

main.py

@@ -7,7 +7,7 @@ import tempfile
 
 #app = Flask(__name__)
 
-app = Flask(__name__, static_folder='static')
+app = Flask(__name__, static_folder='processed_images')
 
 CORS(app, supports_credentials=True)
 #CORS(app, supports_credentials=True, resources={r"/*": {"origins": "*"}})  # Allow requests from any originorigins=["http://localhost:3000"])
@@ -43,8 +43,8 @@ def process_images():
     user_image = temp_path_1
 
     gen_vton(user_image, dress_image)
-    processed_image_1_path = './processed-images/output_image.jpg'
-    processed_image_2_path = './processed-images/output_image.jpg'
+    processed_image_1_path = './processed_images/output_image.jpg'
+    processed_image_2_path = './processed_images/output_image_1.jpg'
     
     # Save your images using the paths above...
 
@@ -58,19 +58,14 @@ def process_images():
     os.remove(temp_path_1)
     return jsonify({'processedImages': processed_image_urls})
 
-@app.route('/processed_images')
-def processed_images():
+@app.route('/processed_images/<filename>')
+def processed_images(filename):
+    print("request_came_here")
+    return send_from_directory(app.static_folder, filename)
     # Example of generating a unique filename for the output
-    import uuid
-    output_filename = f"{uuid.uuid4()}.jpg"
-    processed_image_path = os.path.join('processed-images', output_filename)
-
-# After processing and saving your image to processed_image_path
-    url_to_processed_image = request.host_url + 'static/' + output_filename
-
-    return jsonify({'processedImages': [url_to_processed_image]})
+   
 
-#    return send_from_directory(app.static_folder, filename)
+#    
 
 if __name__ == '__main__':
-    app.run()
+    app.run(debug=True, host='0.0.0.0')