Update app.py

app.py

@@ -11,9 +11,13 @@ MODELS = [
     "Meta-Llama-3.1-8B-Instruct"
 ]
 
-def create_client(api_key):
-    openai.api_key = api_key
-    openai.api_base = "https://api.sambanova.ai/v1"  # Fixed Base URL
+def create_client(api_key = None):
+    if api_key is not None:
+        openai.api_key = api_key
+        openai.api_base = "https://api.sambanova.ai/v1"  # Fixed Base URL
+    else:
+        openai.api_key = os.getenv("API_KEY")
+        openai.api_base = os.getenv("URL")
 
 def chat_with_ai(message, chat_history, system_prompt):
     messages = [