fix output

app.py

@@ -1,26 +1,12 @@
 import gradio as gr
-import random
-from PIL import Image  # Make sure PIL or Pillow is installed
+from PIL import Image, ImageDraw
 
 def generate_image(prompt):
-    try:
-        # Generate a random seed for each request
-        random_seed = random.randint(0, 10000)
-        
-        # Generate image with the model using a random seed
-        image = model.generate(prompt, seed=random_seed)
-        
-        # Ensure the image is in a format compatible with Gradio
-        if isinstance(image, Image.Image):
-            return image
-        else:
-            # If image is not in PIL format, convert or handle it
-            return "Error: Image not in expected format"
-    except Exception as e:
-        return f"Error: {e}"
-
-# Load the model
-model = gr.load("models/black-forest-labs/FLUX.1-schnell")
+    # Create a simple image for testing
+    img = Image.new('RGB', (100, 100), color='red')
+    draw = ImageDraw.Draw(img)
+    draw.text((10, 10), prompt, fill='white')
+    return img
 
 # Create Gradio interface
 iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")