Update app.py

app.py

@@ -56,35 +56,22 @@ def process_image_and_text(image, text):
         api_name="/predict"
     )
 
-    try:
-        result = client.predict(image=image_url, text=text, api_name="/predict")
-        
-        # Важная проверка: если API вернул url, то загружаем его
-        if result and 'url' in result:
-            try:
-                response = requests.get(result['url'], stream=True)
-                response.raise_for_status()
-                img = Image.open(response.raw)
-                return img
-            except requests.exceptions.RequestException as e:
-                print(f"Ошибка при загрузке изображения по URL: {e}")
-                return None
-        elif result and 'path' in result:
-            try:
-                return Image.open(result['path'])
-            except FileNotFoundError:
-                print(f"Файл не найден по пути: {result['path']}")
-                return None
-        else:
-            print("API вернул неожиданный результат:", result)
-            return None
-        
-    except (KeyError, TypeError) as e:
-        print(f"Ошибка при обработке результата API: {e}, результат: {result}")
-        return None
-    except Exception as e:  # Добавляем общий обработчик ошибок
-        print(f"Непредвиденная ошибка: {e}")
-        return None
+    if result is None:
+        return gr.update(value=None)  # Важно: обновляем на None, если нет результата
+    elif isinstance(result, Image.Image):
+        return gr.update(value=result)
+    elif isinstance(result, str) and result.startswith("/tmp"):
+        try:
+            return gr.update(value=Image.open(result))
+        except FileNotFoundError:
+            print(f"Файл не найден по пути: {result}")
+            return gr.update(value=None)
+        except Exception as e:
+            print(f"Ошибка при открытии файла: {e}, путь: {result}")
+            return gr.update(value=None)
+    else:
+        print(f"Неизвестный тип результата: {type(result)}, данные: {result}")
+        return gr.update(value=None)
 
 
 # Интерфейс Gradio с использованием gr.Blocks