Update app.py

app.py

@@ -1,9 +1,19 @@
+from dotenv import load_dotenv
+
+import os
+from pathlib import Path
+
 import gradio as gr
 from huggingface_hub import InferenceClient
 from openai import OpenAI
+
 from prompt_template import PromptTemplate, PromptLoader
 from assistant import AIAssistant
-from pathlib import Path
+
+# Load .env file
+load_dotenv()
+
+API_KEY = os.getenv('API_KEY')
 
 # Load prompts from YAML
 prompts = PromptLoader.load_prompts("prompts.yaml")
@@ -40,6 +50,7 @@ def create_assistant(model_name):
 
     client = OpenAI(
         base_url = "https://integrate.api.nvidia.com/v1",
+        api_key = API_KEY
     )
     
     return AIAssistant(