Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,12 +3,12 @@ from transformers import pipeline
 from gtts import gTTS
 import os
 
-# Load small language model
+# Load a safe, instruction-tuned language model
 @st.cache_resource(show_spinner="Loading language model...")
 def load_llm():
     return pipeline("text-generation",
-                    model="tiiuae/falcon-7b-instruct",
-                    tokenizer="tiiuae/falcon-7b-instruct",
+                    model="mistralai/Mistral-7B-Instruct-v0.1",
+                    tokenizer="mistralai/Mistral-7B-Instruct-v0.1",
                     max_new_tokens=100,
                     do_sample=True,
                     temperature=0.7)
@@ -24,17 +24,20 @@ def speak(text, filename="response.mp3"):
     st.audio(audio_bytes, format="audio/mp3")
     os.remove(filename)
 
-# UI
+# UI setup
 st.set_page_config(page_title="AI Learning Buddy", page_icon="🧸")
-st.title("🧸 AI Learning Buddy for Kids (4–7)")
+st.title("🧸 AI Learning Buddy for Kids (Ages 4–7)")
 
-st.markdown("Ask a question and let the AI Buddy respond in a fun, friendly voice!")
+st.markdown("Ask anything fun or educational — math, animals, colors, or stories!")
 
-user_input = st.text_input("Type your question here (e.g. What is 2 + 3?):")
+# Input field
+user_input = st.text_input("Type your question:")
 
 if st.button("Ask the Buddy") and user_input:
-    prompt = f"You are a friendly teacher for a 5-year-old. Question: {user_input} Answer:"
+    prompt = f"You are a kind, fun teacher for a 5-year-old child. Answer in a simple and cheerful tone.\n\nQuestion: {user_input}\nAnswer:"
     result = llm(prompt)[0]["generated_text"]
     answer = result.split("Answer:")[-1].strip()
+
+    # Output
     st.markdown(f"**AI Buddy says:** {answer}")
     speak(answer)