Spaces:
Sleeping
Sleeping
import gradio as gr | |
import openai | |
import os | |
from dotenv import load_dotenv | |
# 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() | |