Spaces:
Sleeping
Sleeping
import gradio as gr | |
#from langchain.chat_models import ChatOpenAI | |
from langchain_openai import ChatOpenAI | |
from langchain.schema import SystemMessage, HumanMessage | |
# Initialize the model | |
model = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) | |
# Define the PLAN_PROMPT | |
PLAN_PROMPT = "You are an expert writer tasked with writing a high-level outline of a short 3-paragraph essay. Write such an outline for the user-provided topic." | |
# Function to generate the essay outline | |
def generate_essay_outline(topic): | |
messages = [ | |
SystemMessage(content=PLAN_PROMPT), | |
HumanMessage(content=topic) | |
] | |
response = model.invoke(messages) | |
return response.content | |
# Create Gradio interface | |
interface = gr.Interface( | |
fn=generate_essay_outline, # The function to call when the user submits input | |
inputs="text", # The input is a text field (topic) | |
outputs="text", # The output is displayed as text (essay outline) | |
title="Essay Outline Generator", # Optional: Title for the Gradio interface | |
description="Provide a topic, and the model will generate an essay outline for it.", # Description | |
) | |
# Launch the interface | |
interface.launch(share=True) | |