timeout added

Fixing previous bug

app.py

@@ -6,6 +6,7 @@ import gradio as gr
 from transformers import CLIPProcessor, CLIPModel
 import numpy as np
 import time
+import threading
 
 # Replace with your own API key
 STABLE_DIFFUSION_API_KEY = "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"
@@ -85,13 +86,36 @@ def generate_art(mood):
     return image
 
 
+import threading
+
+class ArtGenerationTimeout(Exception):
+    pass
+
+def generate_art_with_timeout(mood, timeout=120):
+    result = {}
+
+    def generate_and_store_result():
+        result['art'] = generate_art(mood)
+
+    thread = threading.Thread(target=generate_and_store_result)
+    thread.start()
+    thread.join(timeout)
+
+    if thread.is_alive():
+        raise ArtGenerationTimeout()
+    else:
+        return result['art']
+
 def mood_art_generator(image):
     mood = get_mood_from_image(image)
     print("Mood:", mood)
     if mood:
-        art = generate_art(mood)
-        output_text = f"You seem to be {mood}. Here's an artwork representing it!"
-        return art, output_text
+        try:
+            art = generate_art_with_timeout(mood)
+            output_text = f"You seem to be {mood}. Here's an artwork representing it!"
+            return art, output_text
+        except ArtGenerationTimeout:
+            return None, "Art generation took too long. Please try again."
     else:
         return None, "Failed to generate artwork."