import gradio as gr
import openai
import os

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

if not api_key:
    print("Warning: OPENAI_API_KEY is not set. Please add it in your Hugging Face Space secrets.")
else:
    openai.api_key = api_key  # Set the API key globally

def chat_with_ai(prompt):
    """Generates a response from OpenAI's GPT-3.5."""
    if not api_key:
        return "Error: Missing API key. Please configure it in the Hugging Face Space settings."

    try:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[{"role": "user", "content": prompt}]
        )
        return response["choices"][0]["message"]["content"]
    except Exception as e:
        return f"Error: {str(e)}"

# Create Gradio interface
iface = gr.Interface(
    fn=chat_with_ai,
    inputs=gr.Textbox(placeholder="Ask me anything..."),
    outputs=gr.Textbox()
)

# Launch the app
if __name__ == "__main__":
    iface.launch()