Update chain_app.py

chain_app.py

@@ -2060,19 +2060,22 @@ async def on_chat_start():
 @cl.on_message  
 async def main(message: cl.Message):
     chat_profile = cl.user_session.get("chat_profile")
-    
+
+    msg = cl.Message(content="")
+    await msg.send()
     if chat_profile == 'neural-brain-AI':
-        completion = openai_client.chat.completions.create(
+        stream = completion = openai_client.chat.completions.create(
             model="ft:gpt-3.5-turbo-1106:nb:aria1:9UWDrLJK",
             messages=[
                 {"role": "system", "content": f"You are neural nexus official chatbot, you are made by Artin Daneshvar and Sadra Noadoust and you are here to help people"},
                 {"role": "user", "content": message.content}
-            ]
+            ],
+            stream=True
         )
-        model_response = completion.choices[0].message.content
-        await cl.Message(
-            content=model_response
-        ).send()
+
+    async for part in stream:
+        if token := part.choices[0].delta.content or "":
+            await msg.stream_token(token)
 
     elif chat_profile == "Dorna-AI":
         result = hf_text_client.predict(