Update app.py

app.py

@@ -34,6 +34,9 @@ preprocess = transforms.Compose([
 
 # Функция предсказания
 def predict(image):
+    if image is None:
+        raise gr.Error("Пожалуйста, загрузите изображение")
+        
     image = image.convert("RGB")
     image_tensor = preprocess(image).unsqueeze(0)
 
@@ -43,20 +46,49 @@ def predict(image):
     
     return {label: float(score) for label, score in zip(labels, scores)}
 
-# Параметры интерфейса
-title = "Skin-AI"
-description = "Sube una imagen"
+# Создаем интерфейс с использованием gr.Blocks
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple", secondary_hue="purple")) as demo:
+    gr.Markdown("# 🩺 Skin-AI")
+    gr.Markdown("### Диагностика кожных заболеваний через анализ изображений")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            image_input = gr.Image(
+                type="pil",
+                label="Загрузите изображение кожи",
+                height=400,
+                tool="editor",
+                show_label=True,
+                show_download_button=True
+            )
+            
+            examples = gr.Examples(
+                examples=["example1.jpg", "example2.jpg", "example3.jpg"],
+                inputs=image_input,
+                label="Примеры изображений",
+                examples_per_page=3
+            )
+            
+        with gr.Column(scale=1):
+            output_label = gr.Label(
+                label="Результат диагностики",
+                num_top_classes=3,
+                show_label=True,
+                show_confidences=True
+            )
+            
+            gr.Markdown("### Рекомендации:")
+            gr.Markdown("- Изображение должно быть четким и хорошо освещенным")
+            gr.Markdown("- Рекомендуется использовать фото без вспышки")
+            gr.Markdown("- Размер файла не должен превышать 10 МБ")
 
-# Создаём интерфейс в стиле, максимально похожем на Spaces
-interface = gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="pil", label="Imagen de entrada"),
-    outputs=gr.Label(num_top_classes=3, label="Predicciones"),
-    title=title,
-    description=description,
-    theme=gr.themes.Soft(primary_hue="purple")  # «Фиолетовая» тема, как на скриншоте
-)
+    image_input.change(
+        fn=predict,
+        inputs=image_input,
+        outputs=output_label,
+        show_progress=True
+    )
 
 # Запуск приложения
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)