Add logging

app.py

@@ -1,3 +1,5 @@
+import logging
+from pathlib import Path
 from time import perf_counter
 
 import gradio as gr
@@ -5,13 +7,14 @@ from jinja2 import Environment, FileSystemLoader
 
 from backend.query_llm import generate
 from backend.semantic_search import qd_retriever
-from pathlib import Path
 
 proj_dir = Path(__file__).parent
-
+# Setting up the logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Set up the template environment with the templates directory
-env = Environment(loader=FileSystemLoader(proj_dir/'templates'))
+env = Environment(loader=FileSystemLoader(proj_dir / 'templates'))
 
 # Load the templates directly from the environment
 template = env.get_template('template.j2')
@@ -36,6 +39,7 @@ def bot(history, system_prompt=""):
     # Create Prompt
     prompt = template.render(documents=documents, query=query)
     prompt_html = template_html.render(documents=documents, query=query)
+    logger.warning(prompt)
 
     history[-1][1] = ""
     for character in generate(prompt, history[:-1]):
@@ -48,7 +52,8 @@ with gr.Blocks() as demo:
         chatbot = gr.Chatbot(
                 [],
                 elem_id="chatbot",
-                avatar_images=('https://aui.atlassian.com/aui/8.8/docs/images/avatar-person.svg', 'https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo.svg'),
+                avatar_images=('https://aui.atlassian.com/aui/8.8/docs/images/avatar-person.svg',
+                               'https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo.svg'),
                 bubble_full_width=False,
                 show_copy_button=True,
                 show_share_button=True,