better handle exceptions

app_dialogue.py

@@ -803,8 +803,11 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
         text_at_the_top,
         generation_args,
     ):
-        time.sleep(random.uniform(0, 1))
-        text = client.generate(prompt=query, **generation_args).generated_text
+        try:
+            text = client.generate(prompt=query, **generation_args).generated_text
+        except Exception as e:
+            logger.error(f"Error {e} while generating meme text")
+            text = ""
         if image is not None and text != "":
             meme_image = make_meme_image(
                 image=image,
@@ -814,6 +817,8 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
                 text_at_the_top=text_at_the_top,
             )
             return meme_image
+        else:
+            return None
 
     def model_inference(
         model_selector,
@@ -904,7 +909,7 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
             ]
 
             for future in concurrent.futures.as_completed(futures):
-                meme_image = future.result()
+                meme_image = future.result(timeout=45)
                 if meme_image:
                     all_meme_images.append(meme_image)
         return user_prompt_str, all_meme_images, chat_history