Update app.py

app.py

@@ -61,16 +61,15 @@ After writing the document, please provide a list of sources used in your respon
     main_content = ""
     try:
         for i in range(num_calls):
-            async for message in client.chat_completion(
+            response = await client.chat_completion(
                 messages=[{"role": "user", "content": prompt}],
                 max_tokens=10000,
                 temperature=temperature,
-                stream=True,
-            ):
-                if message.choices and message.choices[0].delta and message.choices[0].delta.content:
-                    chunk = message.choices[0].delta.content
-                    main_content += chunk
-                    yield main_content, ""
+            )
+            if response.choices and response.choices[0].message:
+                chunk = response.choices[0].message.content
+                main_content += chunk
+                yield main_content, ""
     except Exception as e:
         logging.error(f"Error in get_response_with_search: {str(e)}")
         yield f"An error occurred while processing your request: {str(e)}", ""