Update app.py

app.py

@@ -106,7 +106,27 @@ def query_model(question):
     if not relevant_segment:
         return "Could not find specific information. Please refine your question."
     response = generate_response(question, relevant_segment)
-    return response
+    image = get_image_for_response(question)
+    return response, image 
+    
+IMAGE_DIRECTORY = "Images" 
+def get_image_for_response(question):
+    """
+    Retrieve an image based on the response text.
+    """
+    # Normalize the response text to create a filename
+    file_name = question.lower().replace(" ", "_") 
+    image_path = os.path.join(IMAGE_DIRECTORY, file_name + ".png")
+    print(question)
+    print(image_path)
+    
+    # Check if the image file exists
+    if os.path.exists(image_path):
+        return Image.open(image_path)
+    else:
+        # Return a default or placeholder image if the file is not found
+        default_image_path = os.path.join(IMAGE_DIRECTORY, "default.png")
+        return Image.open(default_image_path)
 
 # Define the welcome message and specific topics the chatbot can provide information about
 welcome_message = """