chatfp / chatfp.py
Federico69's picture
Upload folder using huggingface_hub
c809308 verified
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' )