Inference error handling

app.py

@@ -4,12 +4,20 @@ import gradio as gr
 from gradio.components import Textbox, Button, Slider, Checkbox
 from AinaTheme import theme
 from huggingface_hub import InferenceClient
+from urllib.error import HTTPError
 
 load_dotenv()
 
 def generate(prompt, model_parameters):
-    output = client.text_generation(prompt, **model_parameters, return_full_text=True)
-    return output
+
+    try:
+        output = client.text_generation(prompt, **model_parameters, return_full_text=True)
+        return output
+    except HTTPError as err:
+        if err.code == 400:
+            gr.Warning("The inference endpoint is only available Monday through Friday, from 08:00 to 20:00 CET.")
+    except:
+        gr.Warning('Inference endpoint is not available right now. Please try again later.')        
 
 
 client = InferenceClient(
@@ -38,13 +46,13 @@ def submit_input(input_, max_new_tokens, repetition_penalty, top_k, top_p, do_sa
      
     output = generate(input_, model_parameters)
    
-    if output is  None:
-        gr.Warning('Inference endpoint is not available right now. Please try again later.')
     return output 
     
 def change_interactive(text):
     if len(text.strip()) > MAX_INPUT_CHARACTERS:
         return gr.update(interactive = True),  gr.update(interactive = False)
+    if (len(text) == 0):
+        return gr.update(interactive = True),  gr.update(interactive = False)
     return gr.update(interactive = True),  gr.update(interactive = True)
     
 def clear(): 
@@ -145,6 +153,7 @@ def gradio_app():
                     submit_btn = Button(
                         "Submit", 
                         variant="primary",
+                        interactive=False
                     )
 
         with gr.Row():