Update src/streamlit_app.py

src/streamlit_app.py

@@ -22,10 +22,11 @@ import textwrap
 import requests
 
 # ============================== CONFIGURATION ==============================
-NEO4J_URI = os.getenv["NEO4J_URI"]
-NEO4J_USERNAME = os.getenv["NEO4J_USERNAME"]
-NEO4J_PASSWORD = os.getenv["NEO4J_PASSWORD"]
-openai.api_key = os.getenv["openai_api_key"]
+NEO4J_URI = os.getenv("NEO4J_URI")
+NEO4J_USERNAME = os.getenv("NEO4J_USERNAME")
+NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
 # ============================== DOWNLOAD ==============================
 def download_if_missing(url, local_path):
@@ -265,7 +266,7 @@ Return your response strictly as a **JSON object** with the following fields:
 
 Do not include any explanation or commentary—only return the JSON object.
 """
-    client = openai.OpenAI(api_key=os.getenv["openai_api_key"])
+    client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
     response = client.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "user", "content": prompt}],