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
    )
    return response.choices[0].text.strip()

# Create Gradio interface
input_text = gr.Textbox(label="Enter prompt")
output_text = gr.Textbox(label="AI Response")
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(share=True)