run.py

# This is a simple general-purpose chatbot built on top of LangChain and Gradio.
# Before running this, make sure you have exported your OpenAI API key as an environment variable:
# export OPENAI_API_KEY="your-openai-api-key"

import gradio as gr
from langchain.messages import AIMessage, HumanMessage  # type: ignore
from langchain_openai import ChatOpenAI  # type: ignore

model = ChatOpenAI(model="gpt-4o-mini")


def predict(message, history):
    history_langchain_format = []
    for msg in history:
        if msg["role"] == "user":
            history_langchain_format.append(HumanMessage(content=msg["content"]))
        elif msg["role"] == "assistant":
            history_langchain_format.append(AIMessage(content=msg["content"]))
    history_langchain_format.append(HumanMessage(content=message))
    gpt_response = model.invoke(history_langchain_format)
    return gpt_response.content


demo = gr.ChatInterface(
    predict,
    api_name="chat",
)

if __name__ == "__main__":
    demo.launch()