Update app.py

app.py

@@ -125,10 +125,11 @@ if prompt := st.chat_input():
                     )
                     response.raise_for_status()
 
-                    # Read and yield the response incrementally
-                    for chunk in response.iter_content(chunk_size=256):
-                        if chunk:
-                            yield chunk.decode("utf-8")
+                    # Process and yield only the response field
+                    for line in response.iter_lines():
+                        if line:
+                            data = json.loads(line.decode("utf-8"))
+                            yield data.get("response", "")
                 except requests.exceptions.Timeout:
                     yield "The request timed out. Please try again later."
                 except requests.exceptions.RequestException as e:
@@ -147,4 +148,3 @@ if prompt := st.chat_input():
 
         # Auto-save the updated chat history
         save_chat_history(st.session_state["session_name"], st.session_state["messages"])
-