Update app.py

app.py

@@ -1,16 +1,11 @@
 import gradio as gr
 import os
-from gradio_client import Client
+from gradio_client import Client, APIError
 import time
 import requests
 from deep_translator import GoogleTranslator
 from langdetect import detect
-
-
-MORE = """ ## TRY Other Models
-        ### JARVIS: Your VOICE Assistant -> https://huggingface.co/spaces/KingNish/JARVIS
-        ### Instant Image: 4k images in 5 Second -> https://huggingface.co/spaces/KingNish/Instant-Image
-        """
+import re
 
 # Gradio Client
 client = Client("KingNish/Instant-Video")
@@ -33,10 +28,17 @@ def generate_image(prompt, base="Anime", motion="", step=8, progress=gr.Progress
         )
         video_path = result["video"]
         return video_path
+    except APIError as e:
+        error_message = str(e)
+        match = re.search(r'You have exceeded your GPU quota \((\d+:\d+:\d+) left vs. \d+s requested\). Please retry in (\d+:\d+:\d+)\.', error_message)
+        if match:
+            wait_time = match.group(2)
+            raise gr.Error(f"У модели сработало ограничение (Слишком часто), попробуйте ещё раз в *время: {wait_time}*")
+        else:
+            raise gr.Error("Произошла ошибка при подключении к API. Повторите попытку позже или обратитесь к администраторам Neurix.")
     except ValueError as e:
         print(f"Ошибка подключения.")
         raise gr.Error("Не удалось подключиться к API. Повторите попытку позже или обратитесь к администраторам Neurix.")
-    
 
 # Ссылка на файл CSS
 css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
@@ -124,4 +126,4 @@ with gr.Blocks(css=css) as demo:
         outputs=video,
     )
 
-demo.queue(max_size=250).launch()
+demo.queue(max_size=250).launch()