PTchatKB / app.py
patti-j's picture
Update app.py
732ee73
raw
history blame
958 Bytes
import openai
import gradio as gr
# Initialize OpenAI API key
openai.api_key = "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"
# Load text from file
with open("Dropsheets.txt", "r") as f:
text = f.read()
# Define OpenAI GPT-3.5 model function
def generate_text(prompt):
response = openai.Completion.create(
engine="text-davinci-002",
temperature=0,
max_tokens=1024,
prompt=prompt
)
return response.choices[0].text.strip()
# Create Gradio interface
input_text = gr.Textbox(label="Enter prompt", type="text")
output_text = gr.Textbox(label="AI response", type="text")
demo = gr.Interface(
fn = generate_text,
inputs=input_text,
outputs=output_text,
title="AI Chatbot for PlanetTogether Knowledge Base",
description="Ask a question about the PlanetTogether APS:",
examples=[["How do you create an Alternate Path?"]],
theme="default"
)
# Launch demo
demo.launch()