import os
import gradio as gr
import openai

# Set up OpenAI API
openai.api_key = os.getenv("OPENAI_API_KEY")

def chat_with_gpt(input_text):
    model_engine = "text-davinci-002" # Choose the appropriate model engine.
    prompt = f"{input_text} (AI):"

    response = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        max_tokens=100,
        n=1,
        stop=None,
        temperature=0.8,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0
    )

    generated_text = response.choices[0].text.strip()
    return generated_text

# Gradio user interface
input_text = gr.inputs.Textbox(lines=5, label="Your question about AI:")
output_text = gr.outputs.Textbox(label="ChatGPT Response:")

iface = gr.Interface(fn=chat_with_gpt, inputs=input_text, outputs=output_text, title="Chat with ChatGPT", description="Ask ChatGPT questions about AI.")
iface.launch()