60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from gpt4free import usesless
|
||
import time
|
||
from pywebio import start_server,config
|
||
from pywebio.input import *
|
||
from pywebio.output import *
|
||
from pywebio.session import local
|
||
message_id = ""
|
||
def status():
|
||
try:
|
||
req = usesless.Completion.create(prompt="hello", parentMessageId=message_id)
|
||
print(f"Answer: {req['text']}")
|
||
put_success(f"Answer: {req['text']}",scope="body")
|
||
except:
|
||
put_error("Program Error",scope="body")
|
||
|
||
def ask(prompt):
|
||
req = usesless.Completion.create(prompt=prompt, parentMessageId=local.message_id)
|
||
rp=req['text']
|
||
local.message_id=req["id"]
|
||
print("AI：\n"+rp)
|
||
local.conversation.extend([
|
||
{"role": "user", "content": prompt},
|
||
{"role": "assistant", "content": rp}
|
||
])
|
||
print(local.conversation)
|
||
return rp
|
||
|
||
def msg():
|
||
while True:
|
||
text= input_group("You:",[textarea('You：',name='text',rows=3, placeholder='请输入问题')])
|
||
if not(bool(text)):
|
||
break
|
||
if not(bool(text["text"])):
|
||
continue
|
||
time.sleep(0.5)
|
||
put_code("You："+text["text"],scope="body")
|
||
print("Question："+text["text"])
|
||
with use_scope('foot'):
|
||
put_loading(color="info")
|
||
rp= ask(text["text"])
|
||
clear(scope="foot")
|
||
time.sleep(0.5)
|
||
put_markdown("Bot:\n"+rp,scope="body")
|
||
time.sleep(0.7)
|
||
|
||
@config(title="AIchat",theme="dark")
|
||
def main():
|
||
put_scope("heads")
|
||
with use_scope('heads'):
|
||
put_html("<h1><center>AI Chat</center></h1>")
|
||
put_scope("body")
|
||
put_scope("foot")
|
||
status()
|
||
local.conversation=[]
|
||
local.message_id=""
|
||
msg()
|
||
|
||
print("Click link to chat page")
|
||
start_server(main, port=8099,allowed_origins="*",auto_open_webbrowser=True,debug=True)
|