Update app.py

app.py

@@ -82,10 +82,13 @@ def generate_image(prompt, style, width, height, seed):
             return image
         else:
             logger.error(f"Failed to generate image. Status code: {response.status_code}, Response: {response.text}")
-            return None
+            raise ValueError("Извините, изображение создавалось слишком долго или произошла ошибка. Попробуйте снова.")
+    except requests.exceptions.Timeout:
+        logger.error("Request timed out")
+        raise ValueError("Извините, изображение создавалось слишком долго, мы отменили генерацию.")
     except requests.exceptions.RequestException as e:
         logger.error(f"Request exception: {e}")
-        return None
+        raise ValueError("Произошла ошибка при генерации изображения. Попробуйте снова.")
 
 # Ссылка на файл CSS
 css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
@@ -94,11 +97,10 @@ css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 response = requests.get(css_url)
 css = response.text + ".gradio-container{max-width: 700px !important} h1{text-align:center}"
 
-
-
 # Создаем интерфейс Gradio
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("\n# Генератор Изображений (Lite)")
+    gr.Markdown("")
+    gr.Markdown("# Генератор Изображений (Lite)")
     with gr.Tab("Основные настройки"):
         prompt_input = gr.Textbox(label="Описание изображения", placeholder="Введите описание изображения на английском языке")
         
@@ -154,14 +156,19 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Tab("Дополнительные настройки"):
         with gr.Row():
-            width_slider = gr.Slider(label="Ширина", minimum=1, maximum=4000, value=1024)
-            height_slider = gr.Slider(label="Высота", minimum=1, maximum=4000, value=1024)
+            width_slider = gr.Slider(label="Ширина", minimum=80, maximum=4000, value=1024)
+            height_slider = gr.Slider(label="Высота", minimum=80, maximum=4000, value=1024)
         with gr.Row():
             seed_slider = gr.Slider(label="Сид", minimum=-1, maximum=999999, value=-1, step=1)
     
     create_button = gr.Button("Создать")
     output_image = gr.Image(label="Изображение", type="pil")  # Указываем тип вывода как PIL изображение
+    error_output = gr.Error(label="Ошибка")
 
-    create_button.click(fn=generate_image, inputs=[prompt_input, style_radio, width_slider, height_slider, seed_slider], outputs=output_image)
+    create_button.click(
+        fn=generate_image,
+        inputs=[prompt_input, style_radio, width_slider, height_slider, seed_slider],
+        outputs=[output_image, error_output]
+    )
 
-demo.queue(max_size=150).launch()
+demo.queue(max_size=150).launch()