Update app.py

app.py

@@ -56,8 +56,16 @@ def process_image_and_text(image, text):
         api_name="/predict"
     )
 
-    # Возврат результата
-    return result['path']
+    try:
+        # Обработка результата API (важный момент!)
+        if isinstance(result, dict) and 'path' in result:
+            return result['path']
+        else:
+            return None  # Или другое значение для обработки ошибки
+    except (KeyError, TypeError) as e:
+        print(f"Ошибка при обработке результата API: {e}, результат: {result}")
+        return None
+
 
 # Интерфейс Gradio с использованием gr.Blocks
 with gr.Blocks(css=css) as demo:
@@ -66,7 +74,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             image_input = gr.Image(type="pil", label="Входное изображение")
             text_input = gr.Textbox(lines=2, label="Описание")
-            submit_button = gr.Button("Обработать")
+            submit_button = gr.Button("Обработать", variant="primary")  # Добавили variant='primary'
         with gr.Column():
             output_image = gr.Image(type="pil", label="Результат", show_share_button=False)