Update app.py

app.py

@@ -2,10 +2,10 @@ from flask import Flask, request, jsonify
 from gradio_client import Client, file
 import os
 
-app = Flask(__name__)
+app = Flask(__name__)  # Ensure this line exists
 
 # Initialize Gradio Client
-client = Client("ronniechoyy/IDM-VTON-20250428")
+client = Client("frogleo/AI-Clothes-Changer")
 
 @app.route('/tryon', methods=['POST'])
 def tryon():
@@ -14,33 +14,21 @@ def tryon():
         data = request.get_json()
         
         # Validate required parameters
-        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": file(data.get('background')) if data.get('background') else None,
-            "layers": data.get('layers', []),
-            "composite": file(data.get('composite')) if data.get('composite') else None
-        }
+        if not data or not all(key in data for key in ['person', 'garment']):
+            return jsonify({"error": "Missing required parameters: person, garment"}), 400
 
         # Call Gradio API
         result = client.predict(
-            dict=input_dict,
-            garm_img=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),
+            person=file(data['person']),
+            garment=file(data['garment']),
             denoise_steps=data.get('denoise_steps', 30),
             seed=data.get('seed', 42),
-            num_images=data.get('num_images', 1),
-            api_name="/tryon"
+            api_name="/infer"
         )
 
         # Format response
         response = {
-            "generated_images": result[0],  # List of dictionaries with image and caption
-            "masked_image": result[1]      # Filepath of masked image
+            "generated_image": result  # Filepath of generated image
         }
 
         return jsonify(response), 200