| from poe_api_wrapper import PoeApi |
| from App.Chat.Schemas import BotRequest |
| import re |
|
|
| |
| pattern = r"\!\[.*?\]\((.*?)\)" |
|
|
|
|
| client = PoeApi("sXvCnfYy8CHnXNTRlxhmVg==") |
| CHAT_CODE = "" |
| |
| GEN_CODE = "" |
|
|
|
|
| async def SendMessage(req: BotRequest): |
| global CHAT_CODE, client |
| counter = 0 |
| while True: |
| try: |
| if CHAT_CODE == "": |
| for chunk in client.send_message( |
| req.bot, req.message, chatCode="2rx4w5jt6zf96tn7dr1", file_path=req.file_upload |
| ): |
| pass |
| CHAT_CODE = chunk["chatCode"] |
| else: |
| for chunk in client.send_message( |
| req.bot, req.message, chatCode=CHAT_CODE, file_path=req.file_upload |
| ): |
| pass |
|
|
| return {"response": chunk["text"], "code": 200} |
| except: |
| if counter > 4: |
| return {"response": "Try again later", "code": 500} |
| client = PoeApi("sXvCnfYy8CHnXNTRlxhmVg==") |
| CHAT_CODE = "" |
| counter += 1 |
| print(client.get_chat_history()["data"]) |
|
|
|
|
| async def GenerateImage(req: BotRequest): |
| global GEN_CODE, client |
| counter = 0 |
| while True: |
| try: |
| if GEN_CODE == "": |
| for chunk in client.send_message( |
| req.bot, req.message, chatCode="2rx4w5jt6zf96tn7dr1", file_path=req.file_upload |
| ): |
| pass |
| GEN_CODE = chunk["chatCode"] |
| else: |
| for chunk in client.send_message( |
| req.bot, req.message, chatCode=GEN_CODE, file_path=req.file_upload |
| ): |
| pass |
|
|
| return {"response": re.findall(pattern, chunk["response"])[0], "code": 200} |
| except: |
| if counter > 4: |
| return {"response": "Try again later", "code": 500} |
| client = PoeApi("sXvCnfYy8CHnXNTRlxhmVg==") |
| GEN_CODE = "" |
| counter += 1 |
| client.get_chat_history() |
| |
|
|
