import gradio as gr
import requests
import os

def chat_with_gpt4(prompt):
    api_key = os.getenv("OPENAI_API_KEY")
    if not api_key:
        raise ValueError("API key not found. Please set up the OPENAI_API_KEY environment variable.")

    headers = {
        'Authorization': f'Bearer {api_key}',
        'Content-Type': 'application/json',
    }
    
    data = {
        'model': 'gpt-4',
        'messages': [{'role': 'user', 'content': prompt}],
        'max_tokens': 150
    }
    
    response = requests.post('https://api.openai.com/v1/chat/completions', headers=headers, json=data)
    
    if response.status_code == 200:
        return response.json()['choices'][0]['message']['content']
    else:
        return "Error en la API de OpenAI: " + response.json().get('error', {}).get('message', 'Unknown error')

# Crea la interfaz de Gradio
interface = gr.Interface(
    fn=chat_with_gpt4,
    inputs=gr.Textbox(lines=2, placeholder="Escribe aquí tu pregunta para la IA de FP Dual..."),
    outputs='text',
    title="Chatbot con IA para responder a tus dudas sobre la FP Dual",
    description="Este chatbot utiliza una IA  responder a tus preguntas. ¡Programa en pruebas!. Las respuestas son orientativas"
)

if __name__ == "__main__":
    interface.launch('share=True' )