Update app.py

modified hard coding api values to take from environment

app.py

@@ -26,13 +26,14 @@ from pathlib import Path
 #====================================SETUP=====================================#
 # Fetch secrets from Hugging Face Spaces
 
-model_name = "gpt-4o"
+#model_name = "gpt-4o"
 
 # Extract the OpenAI key and endpoint from the configuration
 openai_key = os.environ["AZURE_OPENAI_KEY"]
 azure_endpoint = os.environ["AZURE_OPENAI_ENDPOINT"]
 api_version = os.environ["AZURE_OPENAI_APIVERSION"]
 
+model_name = os.environ["CHATGPT_MODEL"]
 
 # Define the location of the SQLite database
 db_loc = 'ecomm.db'
@@ -205,6 +206,7 @@ client = AzureOpenAI(
     api_version=os.environ["AZURE_OPENAI_APIVERSION"]
 )
 
+model_name = os.environ["CHATGPT_MODEL"]
 
 @tool
 def rag(user_input: str) -> str:
@@ -232,7 +234,7 @@ def rag(user_input: str) -> str:
     ]
     try:
         response = client.chat.completions.create(
-        model="gpt-4o-mini",
+        model=model_name,
         messages=prompt
         )
 
@@ -549,4 +551,4 @@ def main():
         login_page()
 
 if __name__ == "__main__":
-    main()
+    main()