Update app.py

app.py

@@ -32,9 +32,10 @@ async def v2_chat(prompt: str):
 async def chat_endpoint_get(
     user_prompt: str = Query(..., description="User's prompt"),
     system_prompt: Optional[str] = Query("You are a helpful AI assistant.", description="System prompt to set AI behavior")
-):
+):):
+    ai = CHATv1()
     def generate():
-        ai = CHATv1()
+
         for chunk in ai.chat(user_prompt, system_prompt):
             yield f"data: {chunk}\n\n"
     
@@ -42,8 +43,9 @@ async def chat_endpoint_get(
 
 @app.post("/chatv1")
 async def chat_endpoint_post(request: ChatRequest):
+    ai = CHATv1()
     def generate():
-        ai = CHATv1()
+
         for chunk in ai.chat(request.user_prompt, request.system_prompt):
             yield f"data: {chunk}\n\n"