AI_App / app.py
resolverkatla's picture
update
3bfc696
raw
history blame
1.02 kB
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()