added proper syncs

app.py

@@ -246,22 +246,25 @@ async def bot_comms(message, history):
     gpt_outputs = []
     stream = generation(message, history)
     
-    async for chunk in stream:
-        if chunk is not None:
-            if hasattr(chunk, 'choices') and chunk.choices:
-                delta_content = chunk.choices[0].delta.get("content", "")
-                if delta_content:
-                    gpt_outputs.append(delta_content)
+    try:
+        async for chunk in stream:
+            if chunk is not None:
+                if hasattr(chunk, 'choices') and chunk.choices:
+                    delta_content = chunk.choices[0].delta.get("content", "")
+                    if delta_content:
+                        gpt_outputs.append(delta_content)
+                        yield "".join(gpt_outputs)
+                elif hasattr(chunk, 'image_url'):
+                    # Process image response
+                    image_url = chunk.image_url.get("url", "")
+                    gpt_outputs.append(f"[Image: {image_url}]")
                     yield "".join(gpt_outputs)
-            elif hasattr(chunk, 'image_url'):
-                # Process image response
-                image_url = chunk.image_url.get("url", "")
-                gpt_outputs.append(f"[Image: {image_url}]")
-                yield "".join(gpt_outputs)
+                else:
+                    logger.debug("Chunk does not contain expected attributes.")
             else:
-                logger.debug("Chunk does not contain expected attributes.")
-        else:
-            logger.debug("Chunk is None.")
+                logger.debug("Chunk is None.")
+    except StopAsyncIteration:
+        pass
     # gpt_outputs = []
     # # if mode == "chatting" or mode == "":
     # # logger.debug("On chatting or no mode.\n\n")