Upload 2 files

app.py

@@ -1,20 +1,22 @@
-import gradio as gr
-from gen_tab import create_gen_tab
-from train_tab import create_train_tab
-from dotenv import load_dotenv, find_dotenv
-import os 
-
-_ = load_dotenv(find_dotenv())
-with gr.Blocks(theme=gr.themes.Soft(
-    radius_size=gr.themes.sizes.radius_none,
-    primary_hue=gr.themes.colors.emerald, secondary_hue=gr.themes.colors.green
- 
-                                    )) as demo:
-    with gr.Tabs() as tabs:
-        create_gen_tab()
-        create_train_tab()
-      #  with gr.TabItem("Theme builder"):
-     #       gr.themes.builder()
-      
-
-demo.launch(share=True,debug=True,auth=[("username", "password"),(os.getenv("APP_USER"),os.getenv("APP_PW"))])
+import gradio as gr
+from gen_tab import create_gen_tab
+from train_tab import create_train_tab
+from virtualtryon_tab import create_virtualtryon_tab
+from dotenv import load_dotenv, find_dotenv
+import os 
+
+_ = load_dotenv(find_dotenv())
+with gr.Blocks(theme=gr.themes.Soft(
+    radius_size=gr.themes.sizes.radius_none,
+    primary_hue=gr.themes.colors.emerald, secondary_hue=gr.themes.colors.green
+ 
+                                    )) as demo:
+    with gr.Tabs() as tabs:
+        create_gen_tab()
+        create_train_tab()
+        create_virtualtryon_tab()
+      #  with gr.TabItem("Theme builder"):
+     #       gr.themes.builder()
+      
+
+demo.launch(share=True,debug=True) #,auth=[("username", "password"),(os.getenv("APP_USER"),os.getenv("APP_PW"))])

virtualtryon_tab.py

@@ -0,0 +1,31 @@
+from src.rep_api import virtual_try_on
+import gradio as gr 
+
+
+def create_virtualtryon_tab():
+    with gr.TabItem("Virtual Try-on",id="virtual_tryon"):
+        with gr.Row():
+            with gr.Column():
+                human_img =  gr.Image(label="Human Image")
+                crop = gr.Checkbox(label="Crop", value=False)
+                seed = gr.Number(label="Seed", value=42)
+                steps = gr.Number(label="Steps", value=30)
+            with gr.Column():
+                garm_img = gr.Image(label="Garment Image")
+                category = gr.Dropdown(["upper_body", "lower_body", "full_body"], label="Category", value="upper_body")
+                garment_des = gr.Textbox(label="Garment Description")
+        with gr.Row():
+            tryon_btn = gr.Button("Try-on")
+        with gr.Row():
+            tryon_out = gr.Image(label="Output", type="filepath")
+        with gr.Row():
+           examples = gr.Examples(
+                examples=[
+                    [ False, 42, 30, "upper_body", "Test_images/pink_jumper.png", "Test_images/person_1.jpg","pink jumper"],
+                    [ True, 100, 50, "upper_body", "Test_images/Suit_2.png", "Test_images/person_2.jpg","Suit"],
+                  #  [ False, 7, 20, "full_body", "path/to/garment_img3.jpg", "path/to/human_img3.jpg","Black Suit"]
+                ],
+                inputs=[ crop, seed, steps, category, garm_img,human_img, garment_des]
+            )
+
+    tryon_btn.click(virtual_try_on, inputs=[crop, seed, steps, category,  garm_img, human_img,garment_des],outputs=tryon_out)