Update app.py

app.py

@@ -8,6 +8,37 @@ import requests
 import json
 import time
 
+# Önce dizinleri kontrol et ve gerekirse oluştur
+example_path = os.path.join(os.path.dirname(__file__), 'assets')
+
+# cloth dizini yoksa oluştur
+cloth_path = os.path.join(example_path, "cloth")
+if not os.path.exists(cloth_path):
+    os.makedirs(cloth_path, exist_ok=True)
+    print(f"'{cloth_path}' dizini oluşturuldu.")
+
+# human dizini yoksa oluştur
+human_path = os.path.join(example_path, "human")
+if not os.path.exists(human_path):
+    os.makedirs(human_path, exist_ok=True)
+    print(f"'{human_path}' dizini oluşturuldu.")
+
+# Örnek dosya listelerini al (dizin boşsa boş liste döndür)
+try:
+    garm_list = os.listdir(cloth_path)
+    garm_list_path = [os.path.join(cloth_path, garm) for garm in garm_list]
+except FileNotFoundError:
+    garm_list = []
+    garm_list_path = []
+    print("cloth dizini bulunamadı, boş örnek listesi kullanılıyor")
+
+try:
+    human_list = os.listdir(human_path)
+    human_list_path = [os.path.join(human_path, human) for human in human_list]
+except FileNotFoundError:
+    human_list = []
+    human_list_path = []
+    print("human dizini bulunamadı, boş örnek listesi kullanılıyor")
 
 def tryon(person_img, garment_img, seed, randomize_seed):
     post_start_time = time.time()
@@ -150,14 +181,6 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
 
 MAX_SEED = 999999
 
-example_path = os.path.join(os.path.dirname(__file__), 'assets')
-
-garm_list = os.listdir(os.path.join(example_path,"cloth"))
-garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
-
-human_list = os.listdir(os.path.join(example_path,"human"))
-human_list_path = [os.path.join(example_path,"human",human) for human in human_list]
-
 css="""
 #col-left {
     margin: 0 auto;
@@ -181,9 +204,12 @@ css="""
 """
 
 def load_description(fp):
-    with open(fp, 'r', encoding='utf-8') as f:
-        content = f.read()
-    return content
+    try:
+        with open(fp, 'r', encoding='utf-8') as f:
+            content = f.read()
+        return content
+    except FileNotFoundError:
+        return "<h1>Virtual Try-On Application</h1>"
 
 def change_imgs(image1, image2):
     return image1, image2
@@ -211,7 +237,7 @@ with gr.Blocks(css=css) as Tryon:
             gr.HTML("""
             <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 20px;">
                 <div>
-                Step 3. Press “Run” to get try-on results
+                Step 3. Press "Run" to get try-on results
                 </div>
             </div>
             """)
@@ -219,18 +245,20 @@ with gr.Blocks(css=css) as Tryon:
         with gr.Column(elem_id = "col-left"):
             imgs = gr.Image(label="Person image", sources='upload', type="numpy")
             # category = gr.Dropdown(label="Garment category", choices=['upper_body', 'lower_body', 'dresses'],  value="upper_body")
-            example = gr.Examples(
-                inputs=imgs,
-                examples_per_page=12,
-                examples=human_list_path
-            )
+            if human_list_path:  # Sadece örnekler varsa göster
+                example = gr.Examples(
+                    inputs=imgs,
+                    examples_per_page=12,
+                    examples=human_list_path
+                )
         with gr.Column(elem_id = "col-mid"):
             garm_img = gr.Image(label="Garment image", sources='upload', type="numpy")
-            example = gr.Examples(
-                inputs=garm_img,
-                examples_per_page=12,
-                examples=garm_list_path
-            )
+            if garm_list_path:  # Sadece örnekler varsa göster
+                example = gr.Examples(
+                    inputs=garm_img,
+                    examples_per_page=12,
+                    examples=garm_list_path
+                )
         with gr.Column(elem_id = "col-right"):
             image_out = gr.Image(label="Result", show_share_button=False)
             with gr.Row():
@@ -245,11 +273,8 @@ with gr.Blocks(css=css) as Tryon:
             with gr.Row():
                 seed_used = gr.Number(label="Seed used")
                 result_info = gr.Text(label="Response")
-            # try_button = gr.Button(value="Run", elem_id="button")
             test_button = gr.Button(value="Run", elem_id="button")
 
-
-    # try_button.click(fn=start_tryon, inputs=[imgs, garm_img, seed, randomize_seed], outputs=[image_out, seed_used, result_info], api_name='tryon',concurrency_limit=10)
     test_button.click(fn=tryon, inputs=[imgs, garm_img, seed, randomize_seed], outputs=[image_out, seed_used, result_info], api_name=False, concurrency_limit=45)
 
     with gr.Column(elem_id = "col-showcase"):
@@ -272,4 +297,4 @@ with gr.Blocks(css=css) as Tryon:
             label=None
         )
 
-Tryon.queue(api_open=False).launch(show_api=False)
+Tryon.queue(api_open=False).launch(show_api=False)