Update main.py

main.py

@@ -37,10 +37,12 @@ async def run_all(messages):
 async def clean_chat_history():
     while True:
         current_time = datetime.now()
-        for session_id, history in list(chat_history.items()):
-            if history[-1]['timestamp'] + timedelta(minutes=10) < current_time:
+        # Use list() to create a copy of keys to avoid "dictionary changed size during iteration" issue
+        for session_id in list(chat_history.keys()):
+            if chat_history[session_id][-1]['timestamp'] + timedelta(minutes=10) < current_time:
                 del chat_history[session_id]
         await asyncio.sleep(60)
+
 #-------------------------------------------------------------------------------
 @app.get("/gpt")
 async def chat(request: Request, prompt: str, characteristic: str = None, id: str = None):
@@ -70,17 +72,17 @@ async def chat(request: Request, prompt: str, characteristic: str = None, id: st
         else:
             messages = chat_history[id]
             messages.append({"role": "user", "content": prompt})
-            
-       # run after adds
-       messages = chat_history[id]
-       responses = await run_all(messages)
-            if responses[0]:
-                asyncio.create_task(clean_chat_history())
-                messages.append({"role": "assistant", "content": responses[0]})
-                return {"response": responses[0]}
-            else:
-                asyncio.create_task(clean_chat_history())
-                raise HTTPException(status_code=500, detail="Provider failed")
+
+        # executed after adding from else or if
+        messages = chat_history[id]
+        responses = await run_all(messages)
+        if responses[0]:
+            asyncio.create_task(clean_chat_history())
+            messages.append({"role": "assistant", "content": responses[0]})
+            return {"response": responses[0]}
+        else:
+            asyncio.create_task(clean_chat_history())
+            raise HTTPException(status_code=500, detail="Provider failed")
     except Exception as e:
         raise HTTPException(status_code=401, detail=f"Internal Error: {str(e)}")
 
@@ -94,4 +96,4 @@ async def history_chat(id: str):
     if id not in chat_history:
         raise HTTPException(status_code=404, detail="Session ID not found")
     
-    return {"history": chat_history[id]}
+    return {"history": chat_history[id]}