Reverting back due to no image being produced

app.py

@@ -6,7 +6,6 @@ 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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"
@@ -45,10 +44,6 @@ def get_mood_from_image(image: Image.Image):
 
     return selected_mood
 
-def is_black_image(image: Image.Image) -> bool:
-    img_array = np.array(image)
-    return np.all(img_array == 0)
-
 def generate_art(mood):
     # Implement art generation logic using the Stable Diffusion API
     prompt = f"{mood} generative art with vibrant colors and intricate patterns ({str(np.random.randint(1, 10000))})"
@@ -63,59 +58,33 @@ def generate_art(mood):
     }
 
     while True:
-        while True:
-            response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data)
-
-            if response.status_code == 503:
-                print("Model is loading, waiting for 30 seconds before retrying...")
-                time.sleep(30)
-                continue
-
-            if response.status_code != 200:
-                print(f"Error: API response status code {response.status_code}")
-                print("Response content:")
-                print(response.content)
-                return None
-
-            image = Image.open(BytesIO(response.content))
-            
-            # Check if the image is black
-            if not is_black_image(image):
-                break
-
-    return image
+        response = requests.post('https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5', headers=headers, json=json_data)
 
+        if response.status_code == 503:
+            print("Model is loading, waiting for 30 seconds before retrying...")
+            time.sleep(30)
+            continue
 
-import threading
+        if response.status_code != 200:
+            print(f"Error: API response status code {response.status_code}")
+            print("Response content:")
+            print(response.content)
+            return None
 
-class ArtGenerationTimeout(Exception):
-    pass
+        break
 
-def generate_art_with_timeout(mood, timeout=120):
-    result = {}
+    image = Image.open(BytesIO(response.content))
 
-    def generate_and_store_result():
-        result['art'] = generate_art(mood)
-
-    thread = threading.Thread(target=generate_and_store_result)
-    thread.start()
-    thread.join(timeout)
+    return image
 
-    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:
-        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."
+        art = generate_art(mood)
+        output_text = f"You seem to be {mood}. Here's an artwork representing it!"
+        return art, output_text
     else:
         return None, "Failed to generate artwork."