Update app.py

app.py

@@ -16,9 +16,10 @@ gradio_client.utils.json_schema_to_python_type = lambda schema, defs=None: "stri
 load_dotenv()
 AZURE_OPENAI_API_KEY = os.getenv("AZURE_OPENAI_API_KEY")
 AZURE_OPENAI_ENDPOINT = os.getenv("AZURE_OPENAI_ENDPOINT")
-AZURE_OPENAI_DEPLOYMENT_NAME = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")
+AZURE_OPENAI_LLM_DEPLOYMENT = os.getenv("AZURE_OPENAI_LLM_DEPLOYMENT")
+AZURE_OPENAI_EMBEDDING_DEPLOYMENT = os.getenv("AZURE_OPENAI_EMBEDDING_DEPLOYMENT")
 
-if not all([AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_DEPLOYMENT_NAME]):
+if not all([AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_LLM_DEPLOYMENT, AZURE_OPENAI_EMBEDDING_DEPLOYMENT]):
     raise ValueError("Azure OpenAI environment variables missing.")
 
 # Suppress warnings
@@ -26,7 +27,7 @@ warnings.filterwarnings("ignore")
 
 # Initialize embedding model
 embeddings = AzureOpenAIEmbeddings(
-    azure_deployment=AZURE_OPENAI_DEPLOYMENT_NAME,
+    azure_deployment=AZURE_OPENAI_EMBEDDING_DEPLOYMENT,
     azure_endpoint=AZURE_OPENAI_ENDPOINT,
     openai_api_key=AZURE_OPENAI_API_KEY,
     openai_api_version="2024-08-01-preview",
@@ -41,7 +42,7 @@ vectorstore = FAISS.load_local(
 
 # Initialize LLM
 llm = AzureChatOpenAI(
-    deployment_name=AZURE_OPENAI_DEPLOYMENT_NAME,
+    deployment_name=AZURE_OPENAI_LLM_DEPLOYMENT,
     azure_endpoint=AZURE_OPENAI_ENDPOINT,
     openai_api_key=AZURE_OPENAI_API_KEY,
     openai_api_version="2024-08-01-preview",