Update app.py

app.py

@@ -1,9 +1,7 @@
 import streamlit as st
-from PIL import Image, ImageDraw, ImageColor
-import requests
-from io import BytesIO
-from transformers import pipeline
+from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
 import pyttsx3
+import random
 
 # Set up the app
 st.set_page_config(page_title="ChiliDron", layout="wide", page_icon="🎨")
@@ -11,10 +9,17 @@ st.set_page_config(page_title="ChiliDron", layout="wide", page_icon="🎨")
 # Load Hugging Face models using cache
 @st.cache_resource
 def load_storytelling_model():
-    return pipeline("text-generation", model="gpt2")
+    return pipeline("text-generation", model="bookbot/gpt2-indo-medium-kids-stories")
+
+@st.cache_resource
+def load_phoneme_model():
+    model = AutoModelForSeq2SeqLM.from_pretrained("mirfan899/kids_phoneme_sm_model")
+    tokenizer = AutoTokenizer.from_pretrained("mirfan899/kids_phoneme_sm_model")
+    return model, tokenizer
 
 # Load models
 story_generator = load_storytelling_model()
+phoneme_model, phoneme_tokenizer = load_phoneme_model()
 
 # Initialize TTS engine
 def initialize_tts():
@@ -31,7 +36,7 @@ st.subheader("A fun, interactive, and educational app for kids!")
 
 # Sidebar navigation
 st.sidebar.title("Navigate")
-options = st.sidebar.radio("Go to", ["Interactive Storyteller", "Digital Art Studio", "Guided Meditation"])
+options = st.sidebar.radio("Go to", ["Interactive Storyteller", "Joke and Fun Facts Generator", "Phoneme Practice", "Guided Meditation"])
 
 # Interactive Storyteller
 if options == "Interactive Storyteller":
@@ -45,50 +50,48 @@ if options == "Interactive Storyteller":
     if st.button("Create Story"):
         with st.spinner("Generating your story..."):
             prompt = f"Once upon a time, {character} was {setting} when they {plot}. And then,"
-            story = story_generator(prompt, max_length=200, num_return_sequences=1)
+            story = story_generator(prompt, max_length=500, num_return_sequences=1)
             st.success(story[0]['generated_text'])
 
-# Digital Art Studio
-elif options == "Digital Art Studio":
-    # Define the image URL
-    image_url = "https://huggingface.co/spaces/mohAhmad/ChildDron/raw/main/26141935e868bc1fd22c44de3dc463df.jpg"
-    
-    # Load the image from the URL
-    response = requests.get(image_url)
-    img = Image.open(BytesIO(response.content)).convert("RGBA")
-    
-    # Display the original image
-    st.header("🎨 Cartoon Coloring Studio")
-    st.write("Color the cartoon image!")
-    
-    # Color picker
-    color = st.color_picker("Pick a color", "#000000")
-    
-    # Create a drawing canvas on top of the image
-    canvas_width, canvas_height = img.size
-    drawing_img = img.copy()
-    draw = ImageDraw.Draw(drawing_img)
-    
-    # Display the image
-    st.image(img, caption='Cartoon Image', use_column_width=True)
-    
-    # Function to color the image
-    def color_image(x, y, color):
-        if 0 <= x < canvas_width and 0 <= y < canvas_height:
-            draw.point((x, y), fill=ImageColor.getrgb(color))
-    
-    # Input for coloring
-    click_x = st.slider("Click X", 0, canvas_width - 1, step=1)
-    click_y = st.slider("Click Y", 0, canvas_height - 1, step=1)
-    
-    if st.button("Color"):
-        color_image(click_x, click_y, color)
-        st.image(drawing_img, caption='Your Colored Cartoon', use_column_width=True)
-    
-    # Optionally, save the colored image
-    if st.button("Save Drawing"):
-        drawing_img.save("colored_cartoon.png")
-        st.success("Drawing saved as colored_cartoon.png!")
+# Joke and Fun Facts Generator
+elif options == "Joke and Fun Facts Generator":
+    st.header("🤣 Joke and Fun Facts Generator")
+    st.write("Get ready to laugh and learn with random jokes and fun facts!")
+
+    jokes = [
+        "Why did the scarecrow win an award? Because he was outstanding in his field!",
+        "Why don’t scientists trust atoms? Because they make up everything!",
+        "What do you get if you cross a vampire with a snowman? Frostbite!"
+    ]
+
+    fun_facts = [
+        "Did you know that honey never spoils? Archaeologists have found pots of honey in ancient Egyptian tombs that are over 3,000 years old and still perfectly edible.",
+        "A day on Venus is longer than a year on Venus! It takes about 243 Earth days for Venus to rotate once, but only 225 Earth days to orbit the Sun.",
+        "Octopuses have three hearts and blue blood!"
+    ]
+
+    if st.button("Tell me a joke!"):
+        st.write(random.choice(jokes))
+
+    if st.button("Tell me a fun fact!"):
+        st.write(random.choice(fun_facts))
+
+# Phoneme Practice
+elif options == "Phoneme Practice":
+    st.header("🔤 Phoneme Practice")
+    st.write("Learn how to pronounce words correctly!")
+
+    word_to_practice = st.text_input("Enter a word to practice its phonemes:")
+
+    if st.button("Get Phonemes"):
+        if word_to_practice:
+            with st.spinner("Analyzing phonemes..."):
+                inputs = phoneme_tokenizer.encode(word_to_practice, return_tensors="pt")
+                outputs = phoneme_model.generate(inputs)
+                phonemes = phoneme_tokenizer.decode(outputs[0], skip_special_tokens=True)
+                st.write(f"Phonemes for '{word_to_practice}': {phonemes}")
+        else:
+            st.warning("Please enter a word to practice.")
 
 # Guided Meditation for Kids
 elif options == "Guided Meditation":
@@ -98,12 +101,12 @@ elif options == "Guided Meditation":
     meditation_text = "Close your eyes and take a deep breath. Imagine you are in a peaceful forest, with birds singing and leaves rustling in the wind."
 
     if st.button("Play Meditation"):
-        st.spinner("Playing meditation audio...")
-        tts_engine = initialize_tts()
-        if tts_engine:
-            tts_engine.save_to_file(meditation_text, 'meditation.mp3')
-            tts_engine.runAndWait()
-            st.audio('meditation.mp3', format='audio/mp3')
+        with st.spinner("Playing meditation audio..."):
+            tts_engine = initialize_tts()
+            if tts_engine:
+                tts_engine.save_to_file(meditation_text, 'meditation.mp3')
+                tts_engine.runAndWait()
+                st.audio('meditation.mp3', format='audio/mp3')
 
 # Footer
 st.markdown("<footer style='text-align: center; padding-top: 20px;'>Made with ❤️ for Kids by ChiliDron Team</footer>", unsafe_allow_html=True)