Update app.py

app.py

@@ -45,7 +45,7 @@ def virtual_tryon(human_img_path, garment_img_path, garment_type):
                 "tune[title]": f"Gradio VTO - {int(time.time())}",
                 "tune[name]": garment_type,
                 "tune[model_type]": "faceid",
-                "tune[base_tune_id]": "1504944",
+                "tune[base_tune_id]": "1504944", # This is the base model ID for Virtual Try-On
                 "tune[prompts_attributes][][text]": f"",
                 "tune[prompts_attributes][][num_images]": 1,
                 "tune[prompts_attributes][][denoising_strength]": "0.0",
@@ -64,7 +64,6 @@ def virtual_tryon(human_img_path, garment_img_path, garment_type):
                 raise gr.Error(f"Failed to get prompt ID from tune creation. API response: {tune_data}")
 
     except requests.exceptions.RequestException as e:
-        # UPDATED: Add response text to the error message for better debugging.
         error_message = f"API request failed: {e}"
         if hasattr(e, 'response') and e.response is not None:
             error_message += f". Response: {e.response.text}"
@@ -98,11 +97,9 @@ def virtual_tryon(human_img_path, garment_img_path, garment_type):
             else:
                  yield None, "Step 2/3: Model is training. Checking again in 10 seconds..."
 
-
             time.sleep(10)  # Wait before polling again
 
         except requests.exceptions.RequestException as e:
-            # UPDATED: Add response text to the error message for better debugging.
             error_message = f"Failed to poll for status: {e}"
             if hasattr(e, 'response') and e.response is not None:
                 error_message += f". Response: {e.response.text}"
@@ -112,18 +109,81 @@ def virtual_tryon(human_img_path, garment_img_path, garment_type):
 
     raise gr.Error("Generation timed out after 6 minutes. Please try again.")
 
-
 # --- Gradio User Interface Definition ---
-css = ".container { max-width: 730px; margin: auto; padding-top: 1.5rem; }"
+css = ".container { max-width: 800px; margin: auto; padding-top: 1.5rem; }"
+
+# --- Python code snippet for the API docs ---
+code_snippet = """
+import requests
+import os
+import time
+
+# --- Configuration ---
+ASTRIA_API_BASE_URL = "https://api.astria.ai"
+ASTRIA_API_KEY = os.environ.get("ASTRIA_API_KEY") # Recommended: Load from environment
+
+# --- Image and Parameter Setup ---
+human_img_path = "path/to/your/human.jpg"
+garment_img_path = "path/to/your/garment.jpg"
+garment_type = "shirt"  # Can be "shirt", "pants", or "dress"
+
+headers = {"Authorization": f"Bearer {ASTRIA_API_KEY}"}
+
+# --- API Request ---
+with open(human_img_path, "rb") as human_f, open(garment_img_path, "rb") as garment_f:
+    files = [
+        # The garment image is a regular tune image
+        ("tune[images][]", ("garment.jpg", garment_f.read(), "image/jpeg")),
+        # The human image is the input_image for the prompt
+        ("tune[prompts_attributes][][input_image]", ("human.jpg", human_f.read(), "image/jpeg")),
+    ]
+    data = {
+        "tune[title]": f"My VTO API Test - {int(time.time())}",
+        "tune[name]": garment_type,
+        "tune[model_type]": "faceid",
+        "tune[base_tune_id]": "1504944", # VTO Base Model ID
+        "tune[prompts_attributes][][text]": "", # No text prompt needed for VTO
+        "tune[prompts_attributes][][num_images]": 1,
+        "tune[prompts_attributes][][denoising_strength]": "0.0", # Must be 0 for VTO
+    }
+    # Create the tune and prompt in one call
+    response = requests.post(f"{ASTRIA_API_BASE_URL}/tunes", files=files, data=data, headers=headers)
+
+print("--- Initial Response ---")
+print(response.json())
+
+# You would then poll the prompt ID from the response above to get the final image.
+# See this Space's app.py for a full polling example.
+"""
 
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
+    # --- Banner and Main Title ---
+    gr.HTML(
+        """
+        <div style="text-align: center; padding: 20px; background-color: #f0f8ff; border-radius: 10px; margin-bottom: 20px;">
+            <h2 style="color: #333;">Powered by the Astria.ai API</h2>
+            <p style="color: #555; font-size: 1.1em;">This Virtual Try-On demo uses a powerful, single-call API endpoint.
+            <br>
+            You can integrate this functionality directly into your own applications.
+            </p>
+            <a href="https://docs.astria.ai/docs/use-cases/virtual-try-on/" target="_blank" style="text-decoration: none;">
+                <button style="padding: 10px 20px; font-size: 1em; color: white; background-color: #007bff; border: none; border-radius: 5px; cursor: pointer;">
+                    Read the Full API Documentation
+                </button>
+            </a>
+        </div>
+        """
+    )
+    gr.Markdown("# Virtual Fashion Try-On")
     gr.Markdown(
         """
-        # Virtual Fashion Try-On with Astria API
-        Upload a photo of a person and a photo of a piece of clothing to see them wear it.
+        **Instructions:** Upload a photo of a person and a photo of a piece of clothing to see them wear it.
+        
         **Important:** This Space requires an `ASTRIA_API_KEY`. You can get one from [Astria.ai](https://www.astria.ai/) and set it as a secret in the repository settings.
         """
     )
+    
+    # --- Main Application UI ---
     with gr.Row():
         with gr.Column(scale=1):
             human_image = gr.Image(type="filepath", label="Human Image", height=300)
@@ -136,9 +196,19 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
             )
             submit_btn = gr.Button("Generate", variant="primary")
         with gr.Column(scale=1):
-            status_text = gr.Textbox(label="Status", interactive=False, lines=2)
+            status_text = gr.Textbox(label="Status", interactive=False, lines=2, placeholder="Upload images and click Generate...")
             result_image = gr.Image(label="Result", height=615, interactive=False)
 
+    # --- API Usage Code Snippet ---
+    with gr.Accordion("Show API Usage Code (Python)", open=False):
+        gr.Code(
+            value=code_snippet,
+            language="python",
+            label="Python Request Example",
+            interactive=False
+        )
+
+    # --- Event Listener ---
     submit_btn.click(
         fn=virtual_tryon,
         inputs=[human_image, garment_image, garment_type],
@@ -146,4 +216,4 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()