proxy time

App/Chat/PoeChatrouter.py

@@ -1,11 +1,48 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, HTTPException
 from .utils.PoeBot import SendMessage, GenerateImage
 from .Schemas import BotRequest
-
+from aiohttp import ClientSession
+from pydantic import BaseModel
 
 chat_router = APIRouter(tags=["Chat"])
 
 
+class InputData(BaseModel):
+    input: dict
+    version: str = "727e49a643e999d602a896c774a0658ffefea21465756a6ce24b7ea4165eba6a"
+
+
+async def fetch_predictions(data):
+    async with ClientSession() as session:
+        async with session.post(
+            "https://replicate.com/api/predictions", json=data
+        ) as response:
+            return await response.json(), response.status
+
+
+async def fetch_result(id):
+    url = f"https://replicate.com/api/predictions/{id}"
+    async with ClientSession() as session:
+        async with session.get(url) as response:
+            return await response.json(), response.status
+
+
+@chat_router.post("/predictions")
+async def get_predictions(input_data: InputData):
+    data = {
+        "input": input_data.input,
+        "is_training": False,
+        "create_model": "0",
+        "stream": False,
+        "version": input_data.version,
+    }
+    try:
+        predictions, status_code = await fetch_predictions(data)
+        return predictions, status_code
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")
+
+
 @chat_router.post("/chat")
 async def chat(req: BotRequest):
     return await SendMessage(req)