Fix ui

app.py

@@ -155,27 +155,40 @@ print("🔍 Инициализация моделей...")
 models = load_models()
 pipeline = models["pipeline"]
 
-with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 900px}") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container") as demo:
     gr.Markdown("# 🧥 Virtual Try-On")
+    gr.Markdown("Загрузите фото человека и одежды для виртуальной примерки")
     
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Входные данные")
-            person_input = gr.Image(label="Фото человека", type="pil")
-            cloth_input = gr.Image(label="Фото одежды", type="pil")
-            outfit_prompt = gr.Textbox(label="Описание образа (необязательно)")
-            generate_btn = gr.Button("Сгенерировать", variant="primary")
-        
+            person_input = gr.Image(label="Фото человека", type="pil", sources=["upload"])
+            cloth_input = gr.Image(label="Фото одежды", type="pil", sources=["upload"])
+            outfit_prompt = gr.Textbox(label="Описание образа (опционально)", placeholder="Например: man in casual outfit")
+            clothing_prompt = gr.Textbox(label="Описание одежды (опционально)", placeholder="Например: red t-shirt with print")
+            generate_btn = gr.Button("Сгенерировать примерку", variant="primary")
+            
+            gr.Examples(
+                examples=[
+                    ["./test/person2.png", "./test/00008_00.jpg", "man in skirt", "black longsleeve"]
+                ],
+                inputs=[person_input, cloth_input, outfit_prompt, clothing_prompt],
+                label="Примеры для быстрого тестирования"
+            )
+            
         with gr.Column():
-            gr.Markdown("### Результат")
-            output_image = gr.Image(label="Результат примерки")
-            gr.Markdown("Подождите 1-2 минуты для генерации")
-
+            output_image = gr.Image(label="Результат примерки", interactive=False)
+    
     generate_btn.click(
         fn=generate_vton,
-        inputs=[person_input, cloth_input, outfit_prompt],
+        inputs=[person_input, cloth_input, outfit_prompt, clothing_prompt],
         outputs=output_image
     )
+    
+    gr.Markdown("### Инструкция:")
+    gr.Markdown("1. Загрузите четкое фото человека в полный рост\n"
+                "2. Загрузите фото одежды на белом фоне\n"
+                "3. При необходимости уточните описание образа или одежды\n"
+                "4. Нажмите кнопку 'Сгенерировать примерку'")
 
 if __name__ == "__main__":
     demo.queue(max_size=1).launch(