Update app.py

app.py

@@ -5,38 +5,45 @@ from dotenv import load_dotenv
 
 from langchain.chains import ConversationalRetrievalChain
 from langchain_community.vectorstores import FAISS
-from langchain_openai import ChatOpenAI
-from langchain_community.embeddings import OpenAIEmbeddings
+from langchain_community.chat_models import AzureChatOpenAI
+from langchain_community.embeddings import AzureOpenAIEmbeddings
 
 # Patch Gradio bug (schema parsing issue)
 import gradio_client.utils
 gradio_client.utils.json_schema_to_python_type = lambda schema, defs=None: "string"
 
-
-
 # Load environment variables
 load_dotenv()
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY_P") #or os.getenv("OPENAI_API_KEY_ROBO") 
-if not OPENAI_API_KEY:
-    raise ValueError("Missing OPENAI_API_KEY. Please set it in your environment variables.")
+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")
 
+if not all([AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_DEPLOYMENT_NAME]):
+    raise ValueError("Azure OpenAI environment variables missing.")
 
 # Suppress warnings
 warnings.filterwarnings("ignore")
 
 # Initialize embedding model
-embeddings = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
+embeddings = AzureOpenAIEmbeddings(
+    azure_deployment=AZURE_OPENAI_DEPLOYMENT_NAME,
+    openai_api_key=AZURE_OPENAI_API_KEY,
+    openai_api_base=AZURE_OPENAI_ENDPOINT,
+    openai_api_version="2024-08-01-preview"
+)
 
 # Load FAISS vector store
 vectorstore = FAISS.load_local(
     "faiss_index_sysml", embeddings, allow_dangerous_deserialization=True
 )
 
-# Load ChatOpenAI model
-llm = ChatOpenAI(
-    model_name="gpt-4", 
-    temperature=0.5,
-    openai_api_key=OPENAI_API_KEY
+# Initialize LLM
+llm = AzureChatOpenAI(
+    deployment_name=AZURE_OPENAI_DEPLOYMENT_NAME,
+    openai_api_key=AZURE_OPENAI_API_KEY,
+    openai_api_base=AZURE_OPENAI_ENDPOINT,
+    openai_api_version="2024-08-01-preview",
+    temperature=0.5
 )
 
 # Build conversational chain with history