Update app.py

app.py

@@ -1,11 +1,11 @@
 from flask import Flask, request, jsonify
-from gradio_client import Client, file
+from gradio_client import Client, handle_file
 import os
 
-app = Flask(__name__)  # Ensure this line exists
+app = Flask(__name__)
 
 # Initialize Gradio Client
-client = Client("frogleo/AI-Clothes-Changer")
+client = Client("jallenjia/Change-Clothes-AI")
 
 @app.route('/tryon', methods=['POST'])
 def tryon():
@@ -14,21 +14,34 @@ def tryon():
         data = request.get_json()
         
         # Validate required parameters
-        if not data or not all(key in data for key in ['person', 'garment']):
-            return jsonify({"error": "Missing required parameters: person, garment"}), 400
+        if not data or not all(key in data for key in ['background', 'garm_img', 'garment_des']):
+            return jsonify({"error": "Missing required parameters: background, garm_img, garment_des"}), 400
+
+        # Prepare input dictionary
+        input_dict = {
+            "background": handle_file(data['background']) if data.get('background') else None,
+            "layers": data.get('layers', []),
+            "composite": handle_file(data.get('composite')) if data.get('composite') else None,
+            "id": data.get('id')
+        }
 
         # Call Gradio API
         result = client.predict(
-            person=file(data['person']),
-            garment=file(data['garment']),
+            dict=input_dict,
+            garm_img=handle_file(data['garm_img']),
+            garment_des=data['garment_des'],
+            is_checked=data.get('is_checked', True),
+            is_checked_crop=data.get('is_checked_crop', False),
             denoise_steps=data.get('denoise_steps', 30),
-            seed=data.get('seed', 42),
-            api_name="/infer"
+            seed=data.get('seed', -1),
+            category=data.get('category', 'upper_body'),
+            api_name="/tryon"
         )
 
         # Format response
         response = {
-            "generated_image": result  # Filepath of generated image
+            "output_image": result[0],  # Filepath of output image
+            "masked_image": result[1]   # Filepath of masked image
         }
 
         return jsonify(response), 200
@@ -37,4 +50,4 @@ def tryon():
         return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=5000)