Update app.py

app.py

@@ -1,47 +1,64 @@
+
 import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
 from io import BytesIO
-from PIL import Image
-
-# Initialize transformers for each task
-text_generation_pipe = pipeline("text-generation", model="gpt2")
-text_to_speech_pipe = pipeline("text-to-speech", model="google")
-image_generation_pipe = pipeline("image-generation", model="stable-diffusion")
-
-# Function for generating stories, audio, and illustrations
-def generate_story_and_illustration(prompt, character_name):
-    # Generate story text
-    story_text = text_generation_pipe(prompt=f"Once upon a time, {character_name}... {prompt}", max_length=150)[0]["generated_text"]
-
-    # Generate audio from story text
-    tts_audio = gTTS(text=story_text, lang="en")
-    audio_buffer = BytesIO()
-    tts_audio.save(audio_buffer)
-    audio_buffer.seek(0)
-
-    # Generate image based on story
-    image = image_generation_pipe(prompt=story_text)
-
-    # Return generated story, audio, and image
-    return story_text, audio_buffer.read(), image
-
-# Gradio interface for user interaction
-interface = gr.Interface(
-    fn=generate_story_and_illustration,
-    inputs=[
-        gr.Textbox(label="Start your story with..."),
-        gr.Textbox(label="Give your character a name:"),
-    ],
-    outputs=[
-        gr.Textbox(label="Story"),
-        gr.Audio(label="Listen to the story"),
-        gr.Image(label="See the story come alive"),
-    ],
-    title="Storyteller Playground",
-    description="Create amazing stories with the help of AI!",
-    theme="kids",
+from PIL import Image, ImageDraw, ImageFont
+from IPython.display import Audio
+
+# Create a text generation pipeline with GPT-2
+story_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
+
+# Set the context for the storyteller
+messages = [{"role": "system", "content": "You are a magical storyteller, creating wonderful tales for kids. Make them imaginative and full of joy!"}]
+
+# Define the Storyteller function
+def StorytellerGPT(tell_story):
+    messages.append({"role": "user", "content": tell_story})
+
+    # Generate story using Hugging Face's GPT-2
+    story_reply = story_generator(tell_story, max_length=100, num_return_sequences=1)[0]['generated_text']
+
+    messages.append({"role": "assistant", "content": story_reply})
+
+    # Convert text to speech
+    tts = gTTS(text=story_reply, lang='en', slow=False)
+    audio_io = BytesIO()
+    tts.save(audio_io)
+    audio_io.seek(0)
+
+    # Convert text to image
+    image = generate_dynamic_image(story_reply)
+    
+    return story_reply, Audio(data=audio_io.read(), autoplay=True), image
+
+# Function to generate a dynamic image based on the story text
+def generate_dynamic_image(story_text):
+    # Create a blank image
+    image = Image.new("RGB", (500, 300), (255, 255, 255))
+    draw = ImageDraw.Draw(image)
+    
+    # Use a truetype font file, replace "arial.ttf" with the path to your font file
+    font = ImageFont.truetype("arial.ttf", 20)
+    
+    # Write the story text on the image
+    lines = [story_text[i:i+40] for i in range(0, len(story_text), 40)]
+    y_position = 10
+    for line in lines:
+        draw.text((10, y_position), line, font=font, fill=(0, 0, 0))
+        y_position += 30
+
+    return image
+
+# Create the Gradio Interface
+demo = gr.Interface(
+    fn=StorytellerGPT,
+    inputs="text",
+    outputs=["text", "audio", "image"],
+    title="📖 Storytelling Magic",
+    description="A magical storyteller app for kids! Type a sentence, and let the app create an enchanting story for you."
 )
 
-# Launch the app
-interface.launch()
+# Launch the Gradio Interface
+demo.launch()
+