import gradio as gr
import openai


#from google.colab import userdata
#OPENAI_API_KEY = userdata.get('OPENAI_API_KEY')

import os
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Fetch OpenAI API Key from Hugging Face Secrets
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

def chatbot(prompt):
    """ GPT-4o Mini Chatbot """
    openai.api_key = OPENAI_API_KEY
    response = openai.ChatCompletion.create(
        model="gpt-4o-mini",
        messages=[{"role": "system", "content": "You are a helpful assistant."},
                  {"role": "user", "content": prompt}]
    )
    return response['choices'][0]['message']['content']

# Create Gradio UI
app = gr.Interface(fn=chatbot, inputs="text", outputs="text", title="GPT-4o Mini Chatbot")

# Run the chatbot
app.launch(debug=True)