import gradio as gr
import openai
import os

# Fetch your OpenAI API key from an environment variable
openai.api_key = os.getenv("OPENAI_API_KEY")

if openai.api_key is None:
    raise ValueError("OPENAI_API_KEY environment variable not set.")

def chat_with_gpt(system_message, user_message):
    conversation = f"System: {system_message}\nUser: {user_message}"
    response = openai.Completion.create(
        model="gpt-3.5-turbo",
        prompt=conversation,
        temperature=0.5,
        max_tokens=500,
        frequency_penalty=0.0,
        presence_penalty=0.0,
    )
    return response.choices[0].text.strip()

# Define the interface
iface = gr.Interface(
    fn=chat_with_gpt,
    inputs=[
        gr.inputs.Textbox(lines=2, placeholder="System Message Here..."),
        gr.inputs.Textbox(lines=5, placeholder="Your Message Here..."),
    ],
    outputs="text",
    title="Chat with GPT-3.5",
    description="This Gradio app lets you chat with OpenAI's GPT-3.5 model. Enter a system message for initial context, and then chat as you would with a human.",
    theme="default",  # or "huggingface" for Hugging Face theme
    allow_flagging="never",
).launch()