Update app.py

app.py

@@ -51,13 +51,17 @@ async def respond(
                         chunk_data = chunk.decode('utf-8')
                         response_json = json.loads(chunk_data)
                         if "choices" in response_json:
-                            delta = response_json["choices"][0].get("delta", {})
-                            if "content" in delta:
-                                content = delta["content"]
-                                yield content
+                            content = response_json["choices"][0]["message"]["content"]
+                            yield content
             except json.JSONDecodeError:
-                # 로그 기록하거나 다른 방법으로 처리
-                pass  # 'continue' 대신 'pass' 사용
+                # Log the error or handle it appropriately
+                pass
+            except StopAsyncIteration:
+                # Handle the case where the stream is prematurely terminated
+                pass
+            finally:
+                # Ensure that the generator is properly closed
+                yield "Stream ended"
 
 theme = "Nymbo/Nymbo_Theme"