WebGPT3 / app.py
Web3Daily's picture
Update app.py
bd78d27
raw
history blame
1.67 kB
import openai
import gradio as gr
import os
from functools import reduce
openai.api_key = os.getenv("OPENAI_API_KEY")
def manage_conversation_history(messages, max_tokens=4096):
total_tokens = sum([len(msg["content"]) for msg in messages])
while total_tokens > max_tokens:
# Remove the oldest user and assistant messages
messages.pop(1)
messages.pop(1)
total_tokens = sum([len(msg["content"]) for msg in messages])
return messages
messages = [{"role": "system", "content": "You are a Web3 and cryptocurrency expert that explains Web3, cryptocurrency, blockchain, and financial terminology in terms so simple even a five year old could understand it. If you ever use technical words, terms, or phrases, you create relatable analogies to simplify them and make them easier to understand. In fact, you always open with an analogy when possible."}]
def CustomChatGPT(user_input):
messages.append({"role": "user", "content": user_input})
managed_messages = manage_conversation_history(messages)
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=managed_messages
)
ChatGPT_reply = response["choices"][0]["message"]["content"]
messages.append({"role": "assistant", "content": ChatGPT_reply})
return ChatGPT_reply
description = "Check for ChatGPT outages <a href=https://status.openai.com/>here.</a>"
demo = gr.Interface(
fn=CustomChatGPT,
inputs=gr.Textbox(label="Ask a question:", placeholder="E.g. What are gas fees? ...Can you simplify that with an analogy?"),
outputs=gr.Textbox(label="Answer:"),
title=""
)
demo.launch(inline=True)