Update app.py

app.py

@@ -167,13 +167,13 @@ Context:
 ]
             
             response_text = ""
-            for chunk in self.llm.chat.completions.create(
+            return self.llm.chat.completions.create(
                 model=model_name,
                 messages=messages,
                 max_tokens=500,
                 # stream=True
-            ):
-                return stream.choices[0].message.content
+            ).choices[0].message.content
+                # return stream.choices[0].message.content
                 # if hasattr(chunk.choices[0].delta, 'content'):
                 #     content = chunk.choices[0].delta.content
                 #     if content is not None: