Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import gradio as gr
 from groq import Groq
 import os
+from PIL import Image, ImageDraw
+import io
 from deep_translator import GoogleTranslator
 from deep_translator import GoogleTranslator  # Import the GoogleTranslator class
 import whisper
@@ -104,11 +106,14 @@ def generate_image_from_prompt(prompt):
         d.text((10, 150), "Image Generation Failed", fill=(255, 255, 255))
         return error_img
 
+    # Debug: Check the type of image_bytes
+    print(f"Received image_bytes type: {type(image_bytes)}")
+
     try:
         image = Image.open(io.BytesIO(image_bytes))
         return image
     except Exception as e:
-        print(f"Error: {e}")
+        print(f"Error while opening image: {e}")
         error_img = Image.new('RGB', (300, 300), color=(255, 0, 0))
         d = ImageDraw.Draw(error_img)
         d.text((10, 150), "Invalid Image Data", fill=(255, 255, 255))