Update app.py

app.py

@@ -162,15 +162,14 @@ def query_pdf():
         """
         ### YOU ARE AN EXELSYS EASYHR GUIDE ASSISTANT:
         ### INSTRUCTIONS:
-        - Identify the most relevant topic based on the user query.
-        - Provide step-by-step guidance for that topic only.
-        - Base your response **only** on the selected topic's content.
-        - Do not use content from other topics.
-        - If no relevant topic is found, respond with: "Not found."
+        ### YOU ARE AN EXELSYS EASYHR GUIDE ASSISTANT:
+        ### INSTRUCTIONS:
+        - Your job is to provide step-by-step guidance for the following user query.
+        - Base your response **only** on the retrieved context from the 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."
-        - Format the response like this:
-        <p>Response text here...</p> \n <a>static/output_images/page_67_img_2.png</a>\n <a>static/output_images/page_45_img_1.jpg</a> 
+        - 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. 
         
         ### USER QUERY:
         {query}
@@ -188,9 +187,10 @@ def query_pdf():
 
     # Convert response to string
     response_text = str(chat_response.content)
-
+    print(response_text)
     # Extract multiple image paths from response
-    image_pattern = r"static/output_images/[\w\d_]+\.png"
+    image_pattern = r"(static/output_images/[\w\d_]+\.(?:png|jpg|jpeg|gif))"
+
     extracted_images = re.findall(image_pattern, response_text)
 
     # Remove image paths from text
@@ -201,11 +201,12 @@ def query_pdf():
         "text": cleaned_text,
         "images": extracted_images  # Store multiple images in a list
     }
-    print(response)
+    
 
     return jsonify(response)
 
 
+
 if __name__ == "__main__":
     app.run(debug=True)