Update app.py

app.py

@@ -57,14 +57,34 @@ def process_image_and_text(image, text):
     )
 
     try:
-        # Обработка результата API (важный момент!)
-        if isinstance(result, dict) and 'path' in result:
-            return result['path']
+        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:
-            return None  # Или другое значение для обработки ошибки
+            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
 
 
 # Интерфейс Gradio с использованием gr.Blocks