Update app.py

app.py

@@ -17,26 +17,27 @@ app = Flask(__name__)
 # Load Model and FAISS Index
 
 
-# AzureOpenAiEndpoint= os.getenv("AZURE_OPENAI_ENDPOINT")
-# AzureOpenAiKey= os.getenv("AZURE_OPENAI_KEY")
-# OpenaiApiVersion= os.getenv("OPENAI_API_VERSION")
-# DeploymentName= os.getenv("DEPLOYMENT_NAME")
+AzureOpenAiEndpoint= os.getenv("AZURE_OPENAI_ENDPOINT")
+AzureOpenAiKey= os.getenv("AZURE_OPENAI_KEY")
+OpenaiApiVersion= os.getenv("OPENAI_API_VERSION")
+DeploymentName= os.getenv("DEPLOYMENT_NAME")
+
 model = SentenceTransformer('./sentence-transformers_all-MiniLM-L6-v2')
 index = faiss.read_index("faiss_index.bin")
-groq_api_key = os.getenv('GROQ_API_KEY')
-model_name = "llama-3.3-70b-versatile"
-
-# llm = AzureChatOpenAI(
-#     openai_api_version= OpenaiApiVersion,  # Change based on your API version
-#     deployment_name = DeploymentName,  # The model deployment name in Azure
-#     openai_api_key= AzureOpenAiKey,  # Use the environment variable
-#     azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],  # Use the environment variable
-# )
-llm = ChatGroq(
-            temperature=0, 
-            groq_api_key=groq_api_key, 
-            model_name=model_name
-        )
+# groq_api_key = os.getenv('GROQ_API_KEY')
+# model_name = "llama-3.3-70b-versatile"
+
+llm = AzureChatOpenAI(
+    openai_api_version= OpenaiApiVersion,  # Change based on your API version
+    deployment_name = DeploymentName,  # The model deployment name in Azure
+    openai_api_key= AzureOpenAiKey,  # Use the environment variable
+    azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],  # Use the environment variable
+)
+# llm = ChatGroq(
+#             temperature=0, 
+#             groq_api_key=groq_api_key, 
+#             model_name=model_name
+#         )
 
 with open("metadata.json") as f:
     metadata = json.load(f)
@@ -162,6 +163,8 @@ def query_pdf():
         """
         ### YOU ARE AN EXELSYS EASYHR GUIDE ASSISTANT:
         ### INSTRUCTIONS:
+        ### YOU ARE AN EXELSYS EASYHR GUIDE ASSISTANT:
+        ### INSTRUCTIONS:
         - Your job is to provide step-by-step guidance for the following user query.
         - Check user query and provide relevant information from the Exelsys easyHR PDF.
         - Only provide image path if they are relevant to or you response and user query.
@@ -171,7 +174,7 @@ def query_pdf():
         - If no relevant information is found, simply respond with: "Not found."
         - If the user greets you (e.g., "Hello", "Hi", "Good morning"), respond politely but keep it brief.
         - If the query is unrelated to Exelsys easyHR, respond with: "I'm here to assist with Exelsys easyHR queries only." 
-        - Put images path after response text.
+        - Put images path after response text. 
         
         ### USER QUERY:
         {query}
@@ -203,7 +206,7 @@ def query_pdf():
         "text": cleaned_text,
         "images": extracted_images  # Store multiple images in a list
     }
-    
+    print(response)
 
     return jsonify(response)