Update app.py

app.py

@@ -1,9 +1,14 @@
 import gradio as gr
 from openai import OpenAI
-from API_KEY import OPENAI_API_KEY
+import os
 import sqlite3
 import base64
 
+# Read API key from environment variable
+OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
+if not OPENAI_API_KEY:
+    raise ValueError("API key not found. Please set the OPENAI_API_KEY environment variable.")
+
 client = OpenAI(api_key=OPENAI_API_KEY)
 
 # Database setup