|
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: |
|
|
|
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) |
|
|
