Update app.py

app.py

@@ -4,6 +4,8 @@ import streamlit as st
 from dotenv import load_dotenv
 from openai import OpenAI
 from elevenlabs import generate, play, set_api_key
+import json
+import requests
 
 # Load API keys
 load_dotenv()
@@ -19,23 +21,42 @@ st.title("Explain Like I'm Five (with Voice!)")
 
 topic = st.text_input("What do you want me to explain like you're five?", "")
 
+def stream_openai_response(payload, headers):
+    with requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload, stream=True) as r:
+        for line in r.iter_lines():
+            if line and line.startswith(b"data: "):
+                yield line[len(b"data: "):].decode()
+
 if st.button("Explain with Voice"):
     if topic:
+        st.subheader("Here's how a five-year-old might understand it:")
+        explanation_box = st.empty()
+        full_explanation = ""
+
         with st.spinner(f"Asking my smart brain to explain '{topic}' simply..."):
             prompt = f"Explain {topic} to a five-year-old using very simple words, short sentences, and a fun example or analogy that a young child can easily understand."
-            response = openai_client.chat.completions.create(
-                model="gpt-4o",
-                messages=[
-                    {"role": "user", "content": prompt}
-                ]
-            )
-            explanation = response.choices[0].message.content
+            headers = {"Authorization": f"Bearer {OPENAI_API_KEY}"}
+            payload = {
+                "model": "gpt-4o",
+                "messages": [{"role": "user", "content": prompt}],
+                "temperature": 0.7,
+                "stream": True
+            }
 
-        st.subheader("Here's how a five-year-old might understand it:")
-        st.info(explanation)
+            for chunk in stream_openai_response(payload, headers):
+                if chunk.strip() == "[DONE]":
+                    break
+                try:
+                    parsed = json.loads(chunk)
+                    delta = parsed['choices'][0]['delta'].get('content', '')
+                    full_explanation += delta
+                    explanation_box.markdown(full_explanation)
+                except json.JSONDecodeError:
+                    st.warning(f"Non-JSON chunk received: {chunk}")
+                    continue
 
         with st.spinner("Generating the voice for the explanation..."):
-            audio = generate(text=explanation, voice="Bella", model="eleven_multilingual_v2")
+            audio = generate(text=full_explanation, voice="Bella", model="eleven_multilingual_v2")
 
         st.subheader("Listen to the explanation:")
         st.audio(audio, format="audio/mpeg")