web access gpt

App/Chat/PoeChatrouter.py

@@ -0,0 +1,11 @@
+from fastapi import APIRouter
+from .utils.PoeBot import SendMessage
+from .Schemas import BotRequest
+
+
+chat_router = APIRouter(tags=["Chat"])
+
+
+@chat_router.post("/chat")
+async def chat(req: BotRequest):
+    return await SendMessage(req)

App/Chat/Schemas.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+
+class BotRequest(BaseModel):
+    message: str
+    bot: str

App/Chat/utils/PoeBot.py

@@ -0,0 +1,23 @@
+from poe_api_wrapper import PoeApi
+from App.Chat.Schemas import BotRequest
+
+# import pprint
+
+
+client = PoeApi("sXvCnfYy8CHnXNTRlxhmVg==")
+CHAT_CODE = ""
+
+
+async def SendMessage(req: BotRequest):
+    global CHAT_CODE
+    if CHAT_CODE == "":
+        for chunk in client.send_message(
+            req.bot, req.message, chatCode="2rx4w5jt6zf96tn7dr1"
+        ):
+            pass
+        CHAT_CODE = chunk["chatCode"]
+    else:
+        for chunk in client.send_message(req.bot, req.message, chatCode=CHAT_CODE):
+            pass
+
+    return chunk["text"]

App/app.py

@@ -1,10 +1,11 @@
 from fastapi import FastAPI
 
 from fastapi.middleware.gzip import GZipMiddleware
+
 from .TTS.TTSRoutes import tts_router
 
 from .Embedding.EmbeddingRoutes import embeddigs_router
-
+from .Chat.PoeChatrouter import chat_router
 
 from fastapi.middleware.cors import CORSMiddleware
 
@@ -47,3 +48,4 @@ async def landing_page():
 
 app.include_router(embeddigs_router)
 app.include_router(tts_router)
+app.include_router(chat_router)