code_test_conversation.py

from EdgeGPT.EdgeUtils import Query, Cookie
from EdgeGPT.EdgeGPT import Chatbot, ConversationStyle
from datetime import datetime
from dotenv import load_dotenv

import telebot
import asyncio
import json
import os

cookies = json.loads(open("cookies.json", encoding="utf-8").read())

async def main():

    bot_ai = await Chatbot.create(cookies = cookies)

    while True:

        prompt = input('Write your prompt: ')
        time_prompt = datetime.now().strftime("%H:%M:%S")

        if prompt.lower() != 'exit':

            response = await bot_ai.ask(prompt = prompt, conversation_style = ConversationStyle.creative, simplify_response = True)
            time_response = datetime.now().strftime("%H:%M:%S")
            conversation = f"{time_prompt} Prompt:\n\n{prompt}\n\n{time_response} Response:\n\n{response['text']}\n\n"
            with open("conversation.md", "a", encoding="utf-8") as file: file.write(conversation)

        else: 
            await bot_ai.close()
            break

if __name__ == "__main__":
    asyncio.run(main())