Update app.py

app.py

@@ -10,7 +10,30 @@ import json
 API_KEY = 'N80HWHVG3DV8URRNYZY382UPSHP1N8G1SNPYG0E9'
 API_URL = 'https://api.runpod.ai/v2/31jyh9kh7nwyga'
 
-# ... [previous code for cloth_images, user_images, and scene_options remains unchanged] ...
+cloth_images = [
+    {"url": "https://i.postimg.cc/7ZzLZtbc/hmgoepprod-4-1.jpg", "label": "Jacket 1"},
+    {"url": "https://i.postimg.cc/7Yd6DrP0/hmgoepprod-6.jpg", "label": "Jacket 2"},
+    {"url": "https://i.postimg.cc/8z40MNFr/hnm.png", "label": "Jacket 3"},
+    {"url": "https://i.postimg.cc/mkqbb74B/hmgoepprod-5.jpg", "label": "Jacket 4"},
+    {"url": "https://i.postimg.cc/vBwySGzH/hmgoepprod.jpg", "label": "Jacket 5"},
+    {"url": "https://i.postimg.cc/6qSYYprM/Whats-App-Image-2024-08-18-at-13-45-37-2.jpg", "label": "Jacket 6"},
+    {"url": "https://i.postimg.cc/tJHhTnzQ/Screenshot-2024-08-20-at-10-04-29-PM.png", "label": "Jacket 7"},
+    {"url": "https://i.postimg.cc/yxQYGwxQ/Screenshot-2024-08-20-at-10-06-28-PM.png", "label": "Jacket 8"},
+]
+
+user_images = [
+    {"url": "https://iili.io/dEk7qtp.md.jpg", "label": "User 1 (Male)"},
+    {"url": "https://iili.io/d1RpKBa.md.jpg", "label": "User 2 (Female)"},
+    {"url": "https://i.postimg.cc/QtzQrnFt/Whats-App-Image-2024-08-18-at-13-42-10.jpg", "label": "User 3 (Female)"},
+    {"url": "https://i.postimg.cc/rsKLDzyj/Whats-App-Image-2024-08-19-at-12-13-41.jpg", "label": "User 4 (Female)"},
+    {"url": "https://i.postimg.cc/cHkpGZ7n/Whats-App-Image-2024-08-19-at-00-27-16-1.jpg", "label": "User 5 (Female)"},
+    {"url": "https://i.postimg.cc/g2bfWghN/Whats-App-Image-2024-08-19-at-12-13-41-1.jpg", "label": "User 6 (Female)"},
+]
+
+scene_options = [
+    "Gym", "City Street", "Party", "Beach", "Office", "Park", "Cafe", 
+    "Shopping Mall", "Concert", "Sports Stadium"
+]
 
 def get_base64_from_url(url):
     response = requests.get(url)
@@ -38,21 +61,13 @@ def generate_tryon(cloth_input, user_input, background):
         "background": background
     }
 
-    # Prepare log message
-    log_data = {
-        "user_image": user_base64[:20] + '...' if user_base64 else 'undefined',
-        "product_image": cloth_base64[:20] + '...' if cloth_base64 else 'undefined',
-        "background": background,
-    }
-    log_message = f"Sending data:\n{json.dumps(log_data, indent=2)}"
-
     response = requests.post(
         f"{API_URL}/run",
         headers={
             "Authorization": f"Bearer {API_KEY}",
             "Content-Type": "application/json"
         },
-        json=input_data
+        data=json.dumps({"input": input_data})
     )
 
     if not response.ok:
@@ -74,7 +89,7 @@ def generate_tryon(cloth_input, user_input, background):
         if status_data['status'] == 'COMPLETED':
             output_base64 = status_data['output']['output']
             output_image = Image.open(io.BytesIO(base64.b64decode(output_base64)))
-            return output_image, log_message
+            return output_image
         elif status_data['status'] == 'FAILED':
             raise Exception(f"Job processing failed: {status_data}")
         
@@ -86,17 +101,17 @@ def tryon_interface(cloth_selected, cloth_upload, user_selected, user_upload, sc
     background = custom_scene if custom_scene else scene_selection
 
     if not cloth:
-        return None, "Please select or upload a clothing image.", ""
+        return None, "Please select or upload a clothing image."
     if not user:
-        return None, "Please select or upload a user image.", ""
+        return None, "Please select or upload a user image."
     if not background:
-        return None, "Please select or enter a background scene.", ""
+        return None, "Please select or enter a background scene."
 
     try:
-        result_image, log_message = generate_tryon(cloth, user, background)
-        return result_image, "Try-on image generated successfully!", log_message
+        result_image = generate_tryon(cloth, user, background)
+        return result_image, "Try-on image generated successfully!"
     except Exception as e:
-        return None, f"Error: {str(e)}", ""
+        return None, f"Error: {str(e)}"
 
 def select_image(evt: gr.SelectData, image_list):
     if evt.index < len(image_list):
@@ -137,7 +152,6 @@ with gr.Blocks() as demo:
     
     output_image = gr.Image(label="Try-On Result")
     output_text = gr.Textbox(label="Status")
-    log_output = gr.Textbox(label="API Request Details", lines=10)
 
     cloth_gallery.select(
         fn=lambda evt: select_image(evt, cloth_images),
@@ -154,7 +168,7 @@ with gr.Blocks() as demo:
     generate_button.click(
         tryon_interface,
         inputs=[cloth_selected, cloth_upload, user_selected, user_upload, scene_selection, custom_scene],
-        outputs=[output_image, output_text, log_output]
+        outputs=[output_image, output_text]
     )
 
 demo.launch()