Update app.py

app.py

@@ -1,71 +1,39 @@
-from flask import Flask, request, jsonify, send_from_directory
-from flask_cors import CORS
-from genai import gen_vton
-from werkzeug.utils import secure_filename
+import streamlit as st
+from PIL import Image
 import os
 import tempfile
+from genai import gen_vton
+from werkzeug.utils import secure_filename
 
-#app = Flask(__name__)
-
-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"])
-
-#CORS(app, resources={r"/proc": {"origins": "http://localhost:3000"}}, supports_credentials=True)
-#@app.route("/proc")
-@app.route('/proc', methods=['POST'])
-def process_images():
-    # Retrieve images from the request
-    print("Request came here")
-    print(request)
-    print(request.headers)
-    print(request.files)
-   
-    
-    user_image_t = request.files.get('userImage')
-    dress_image_t = request.files.get('dressImage')
-    #print(dress_image_t.filename)
-    print(user_image_t.filename)
-    #file = request.files['file']
-    if dress_image_t:
-        # Save the file to a temporary file
-        temp_dir = tempfile.gettempdir()
-        filename = secure_filename(dress_image_t.filename)
-        temp_path = os.path.join(temp_dir, filename)
-        dress_image_t.save(temp_path)
-    dress_image = temp_path
-    if user_image_t:
-        temp_dir = tempfile.gettempdir()
-        filename = secure_filename(user_image_t.filename)
-        temp_path_1 = os.path.join(temp_dir, filename)
-        user_image_t.save(temp_path_1)
-    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_1.jpg'
-    
-    # Save your images using the paths above...
-
-    # Return the URL for the saved images
-    url_to_processed_image_1 = request.host_url + processed_image_1_path
-    url_to_processed_image_2 = request.host_url + processed_image_2_path
-    # Process images...
-    # For the sake of this example, let's say the processing function returns two image URLs
-    processed_image_urls = [url_to_processed_image_1, url_to_processed_image_2]
-    os.remove(temp_path)
-    os.remove(temp_path_1)
-    return jsonify({'processedImages': processed_image_urls})
-
-@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
-   
-
-#    
-
-if __name__ == '__main__':
-    app.run()
+# Define a function to save the uploaded file to a temporary directory
+def save_uploaded_file(uploaded_file):
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix="." + uploaded_file.name.split('.')[1]) as tmp:
+            tmp.write(uploaded_file.getbuffer())
+            return tmp.name
+    except Exception as e:
+        return None
+
+# Streamlit UI components
+st.title('Image Processing with gen_vton')
+
+# File uploader for the user and dress images
+user_image_up = st.file_uploader("Upload User Image", type=['jpg', 'png', 'jpeg'])
+dress_image_up = st.file_uploader("Upload Dress Image", type=['jpg', 'png', 'jpeg'])
+
+# Process the images once both are uploaded and a button is pressed
+if user_image_up and dress_image_up and st.button('Process Images'):
+    user_image_path = save_uploaded_file(user_image_up)
+    dress_image_path = save_uploaded_file(dress_image_up)
+
+    # Assuming gen_vton processes the images and saves them somewhere
+    processed_image_paths = gen_vton(user_image_path, dress_image_path)  # Adjust based on actual return value
+
+    # Displaying processed images (assuming gen_vton returns paths to processed images)
+    if processed_image_paths:
+        for image_path in processed_image_paths:
+            st.image(image_path, use_column_width=True)
+
+    # Clean up temporary files
+    os.remove(user_image_path)
+    os.remove(dress_image_path)