Update main.py

main.py

@@ -3,6 +3,7 @@ from typing import List
 from fastapi import FastAPI, HTTPException, Response, Query, Request
 from typing import Optional
 from pydantic import BaseModel
+from characterai import PyCAI
 import g4f
 import asyncio
 
@@ -43,20 +44,25 @@ async def clean_chat_history():
                 del chat_history[session_id]
         await asyncio.sleep(60)
 #-------------------------------------------------------------------------------
-@app.get("/AnotherAPI/GPT/Noir/{prompt}")
-async def chat(request: Request, prompt: str):
-    session_id = request.client.host
-    if session_id not in chat_history:
-        chat_history[session_id] = [noir_characteristic]
-
-    messages = chat_history[session_id]
-    messages.append({"role": "user", "content": prompt})
-
-    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 to generate a response")
+@app.get("/AnotherAPI/GPT/Noir/{api_key}/{prompt}")
+async def chat(request: Request, api_key: str, prompt: str):
+    try:
+        client = PyCAI(api_key)
+        client.chat.new_chat('csTC3hw0Fnj1Whnl0uV1Nb3_oYIillMQtdBH5NEl0Gs')
+        session_id = request.client.host
+        if session_id not in chat_history:
+            chat_history[session_id] = [noir_characteristic]
+
+        messages = chat_history[session_id]
+        messages.append({"role": "user", "content": prompt})
+
+        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"Invalid API key: {str(e)}")