Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import torch
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 from diffusers import StableDiffusionPipeline
-from PIL import Image, ImageDraw, ImageFont
+from PIL import Image
 
 # Check for GPU availability
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -64,25 +64,25 @@ def generate_story(prompt):
         return "Error generating story."
 
 # Function to add a speech bubble to an image
-def add_speech_bubble(image, text, position=(50, 50)):
-    draw = ImageDraw.Draw(image)
+# def add_speech_bubble(image, text, position=(50, 50)):
+#     draw = ImageDraw.Draw(image)
 
-    try:
-        font = ImageFont.truetype("arial.ttf", 20)
-    except IOError:
-        font = ImageFont.load_default()
+#     try:
+#         font = ImageFont.truetype("arial.ttf", 20)
+#     except IOError:
+#         font = ImageFont.load_default()
 
-    text_bbox = draw.textbbox((0, 0), text, font=font)
-    text_width = text_bbox[2] - text_bbox[0]
-    text_height = text_bbox[3] - text_bbox[1]
+#     text_bbox = draw.textbbox((0, 0), text, font=font)
+#     text_width = text_bbox[2] - text_bbox[0]
+#     text_height = text_bbox[3] - text_bbox[1]
 
-    bubble_width, bubble_height = text_width + 30, text_height + 20
-    bubble_x, bubble_y = position
+#     bubble_width, bubble_height = text_width + 30, text_height + 20
+#     bubble_x, bubble_y = position
 
-    draw.ellipse([bubble_x, bubble_y, bubble_x + bubble_width, bubble_y + bubble_height], fill="white", outline="black")
-    draw.text((bubble_x + 15, bubble_y + 10), text, font=font, fill="black")
+#     draw.ellipse([bubble_x, bubble_y, bubble_x + bubble_width, bubble_y + bubble_height], fill="white", outline="black")
+#     draw.text((bubble_x + 15, bubble_y + 10), text, font=font, fill="black")
 
-    return image
+#     return image
 
 # Streamlit UI
 st.title("🦸‍♂️ AI Comic Story Generator")
@@ -109,8 +109,8 @@ if user_prompt:
 
                 # Extract first sentence (50 characters max) for speech bubble
                 speech_text = generated_story.split(".")[0][:50]
-                image_with_bubble = add_speech_bubble(image, speech_text, position=(50, 50))
+                # image_with_bubble = add_speech_bubble(image, speech_text, position=(50, 50))
 
-                st.image(image_with_bubble, caption="Generated Comic Image", use_container_width=True)
+                st.image(image, caption="Generated Comic Image", use_container_width=True)
             except Exception as e:
                 st.error(f"❌ Error generating image: {e}")