PTchatKB / app.py
patti-j's picture
Update app.py
c7c1b41
raw
history blame
1.36 kB
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()
# Define password-protected Gradio interface
def check_password(password):
if password == "patti-ai":
gr.Interface.visible = True
password_box.visible = False
return "Access granted"
else:
gr.Interface.visible = False
password_box.visible = True
return "Access denied"
# Create Gradio interface
password_box = gr.Textbox(label="Enter password", type="password")
input_text = gr.Textbox(label="Enter prompt", type="text")
output_text = gr.Textbox(label="AI response", type="text")
demo = gr.Interface(
fn=check_password,
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()