Update app.py

app.py

@@ -19,60 +19,80 @@ def get_random_api_key():
 # Ссылка на файл CSS
 css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 
-# Получение CSS по ссылке
+# Получение CSS по ссылке (с обработкой ошибок)
 try:
-    response = requests.get(css_url)
-    response.raise_for_status()  # Поднимаем исключение, если статус ответа не 200
+    response = requests.get(css_url, timeout=10)  # Добавлено timeout
+    response.raise_for_status()
     css = response.text + " h1{text-align:center}"
 except requests.exceptions.RequestException as e:
     print(f"Ошибка при загрузке CSS: {e}")
-    css = " h1{text-align:center}"  # Используем базовый стиль, если загрузка CSS не удалась
+    css = " h1{text-align:center}"  # Используем базовый стиль
 
 # Функция для обработки изображения и текста через API
 def process_image_and_text(image, text):
-    # Проверка и перевод текста на английский, если необходимо
-    language = detect(text)
-    if language != 'en':
-        text = GoogleTranslator(source=language, target='en').translate(text)
-
-    # Сохранение изображения во временный файл
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
-        image.save(temp_file.name)
-        image_path = temp_file.name
-
-    # Загрузка изображения в кеш и получение ссылки
-    image_url = handle_file(image_path)
-
-    # Получение случайного API ключа
-    api_key = get_random_api_key()
-
-    # Инициализация клиента
-    client = Client("Yuanshi/OminiControl", hf_token=api_key)
-
-    # Вызов API
-    result = client.predict(
-        image=image_url,
-        text=text,
-        api_name="/predict"
-    )
+    try:
+        # Проверка и перевод текста на английский, если необходимо
+        language = detect(text)
+        if language != 'en':
+            text = GoogleTranslator(source=language, target='en').translate(text)
 
-    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)
+        # Сохранение изображения во временный файл
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+            image.save(temp_file.name)
+            image_path = temp_file.name
+
+        # Загрузка изображения в кеш и получение ссылки
+        image_url = handle_file(image_path)
+
+        # Получение случайного API ключа
+        api_key = get_random_api_key()
+
+        # Инициализация клиента (обработка ошибок)
+        client = Client("Yuanshi/OminiControl", hf_token=api_key)
+        result = client.predict(image=image_url, text=text, api_name="/predict")
 
+        if result is None:
+            return None
+        elif isinstance(result, Image.Image):
+            return result
+        elif isinstance(result, str) and result.startswith("/tmp"):
+            try:
+                return Image.open(result)
+            except FileNotFoundError:
+                print(f"Файл не найден: {result}")
+                return None
+            except Exception as e:
+                print(f"Ошибка при открытии файла: {e}, путь: {result}")
+                return None
+        else:
+            print(f"Неизвестный тип результата: {type(result)}, данные: {result}")
+            return None
+    except Exception as e:
+        print(f"Ошибка при обработке: {e}")
+        return None
+
+# Функция для получения примеров
+def get_samples():
+    sample_list = [
+        {
+            "image": "img/example_1in.png",
+            "text": "Луна, данный предмет лежит на поверхности луны — груша. На заднем плане Земля возвышается на переднем плане, уль��ра реализм",
+        },
+    ]
+
+    samples_with_images = []
+    for sample in sample_list:
+        image_path = sample["image"]
+        if os.path.exists(image_path):
+            try:
+                img = Image.open(image_path)
+                samples_with_images.append([img, sample["text"]])
+            except Exception as e:
+                print(f"Ошибка при открытии изображения {image_path}: {e}")
+        else:
+            print(f"Файл изображения {image_path} не найден.")
+
+    return samples_with_images
 
 # Интерфейс Gradio с использованием gr.Blocks
 with gr.Blocks(css=css) as demo:
@@ -92,9 +112,14 @@ with gr.Blocks(css=css) as demo:
         concurrency_limit=250
     )
 
+    # Добавление примеров
+    examples = get_samples()
+    if examples:
+        demo.load_examples(examples)
+
 # Запуск интерфейса
 if __name__ == "__main__":
     demo.launch(
         show_api=False,
         share=False
-    )
+    )