Update app.py

app.py

@@ -178,10 +178,14 @@ logger.info("🟢 Agent initialized successfully!")
 # Defines a function to process a user's question.
 def ask_agent(question):
     logger.info(f"Question asked: {question[:100]}...")
-    # Run the agent with the user's question, ensuring it uses its knowledge base.
-    response = agent.run(question, use_knowledge=True)
-    # Get the agent's response as a single string.
-    full_content = response.get_content_as_string()
+    try:
+        # Run the agent with the user's question, ensuring it uses its knowledge base.
+        response = agent.run(question, use_knowledge=True)
+        # Get the agent's response as a single string.
+        full_content = response.get_content_as_string()
+    except Exception as e:
+        logger.error(str(e))
+        return "❌ Something went wrong. Please try again.", None
     # Use a regular expression to find the first URL ending in '.pdf' in the response.
     match = re.search(r'https?://[^\s]+\.pdf', full_content, re.IGNORECASE)
     # Extract the link if a match is found, otherwise set it to None.
@@ -192,6 +196,7 @@ def ask_agent(question):
     else:
         logger.info("🔴 No PDF link found in response")
     # Return the full text response and the extracted PDF link.
+    full_content += "\n\n---\n**🔍 Try asking:**\n- Give me a real example...\n- Explain step by step...\n- Compare with alternatives..."
     return full_content, link
 
 # Defines a function to download the raw content of a PDF from a URL.
@@ -268,6 +273,19 @@ with gr.Blocks(
         with gr.Column(scale=3):
             # The chatbot display window.
             chatbot = gr.Chatbot(label="💬 Conversation", elem_classes="chatbot", height=450)
+            # 👍👎 Feedback buttons
+            with gr.Row():
+                thumbs_up = gr.Button("👍 Helpful")
+                thumbs_down = gr.Button("👎 Not Helpful")
+             
+            # Hidden feedback box (only appears on 👎)
+            feedback_box = gr.Textbox(
+                placeholder="💬 Optional: tell us what went wrong...",
+                visible=False
+            )
+             
+            submit_feedback_btn = gr.Button("Submit Feedback", visible=False)
+            feedback_status = gr.Markdown("")
             # A text area for status messages (used for PDF loading status).
             status_text = gr.Markdown("")
             # The textbox where the user types their question.
@@ -277,7 +295,10 @@ with gr.Blocks(
                 lines=1
             )
             # The submit button.
-            ask_btn = gr.Button("Submit 📤", variant="primary")
+            #ask_btn = gr.Button("Submit 📤", variant="primary")
+            with gr.Row():
+                ask_btn = gr.Button("Submit 📤", variant="primary")
+                clear_btn = gr.Button("🧹 Clear Chat")
             # A section for example questions.
             gr.Markdown("### 💡 Example Questions")
             gr.Examples(
@@ -303,6 +324,7 @@ with gr.Blocks(
                 label="⬇️ Cheat Sheet Preview",
                 visible=False
             )
+            pdf_link_btn = gr.Markdown("")
 
     # Defines the main chat logic as a generator function for streaming output.
     def chat_ui(user_message, chat_history):
@@ -319,7 +341,7 @@ with gr.Blocks(
         # Append a temporary "Thinking..." message from the assistant.
         chat_history.append({
             "role": "assistant",
-            "content": "🤔 Thinking..."
+            "content": "🤔 Dox is thinking..."
         })
     
         # `yield` immediately updates the UI with the user's message and "Thinking...".
@@ -359,6 +381,37 @@ with gr.Blocks(
             [chatbot, link_state, output_image, question]
         )
 
+    def show_pdf_link(link):
+        if link:
+            return f"[📥 Open Full PDF]({link})"
+        return ""
+
+    def clear_chat():
+        return [], None, gr.update(value=None, visible=False)
+
+    clear_btn.click(
+        clear_chat,
+        outputs=[chatbot, link_state, output_image]
+    )
+
+    def show_feedback_box():
+        return gr.update(visible=True), gr.update(visible=True)
+     
+    thumbs_down.click(
+        show_feedback_box,
+        outputs=[feedback_box, submit_feedback_btn]
+    )
+
+    def handle_feedback(text):
+        logger.info(f"User feedback: {text}")
+        return "✅ Feedback submitted. Thank you!"
+     
+    submit_feedback_btn.click(
+        handle_feedback,
+        inputs=feedback_box,
+        outputs=feedback_status
+    )
+
     # Set up the event handler for the "Submit" button click.
     ask_btn.click(
         *submit_chain()
@@ -373,6 +426,10 @@ with gr.Blocks(
         display_pdf,
         inputs=link_state, # Use the same link.
         outputs=[output_image, pdf_status] # Update the image and hide the status text.
+    ).then(
+    show_pdf_link,
+    inputs=link_state,
+    outputs=pdf_link_btn
     )
     
     # Set up the same event handler for when the user presses Enter in the textbox.
@@ -386,6 +443,10 @@ with gr.Blocks(
         display_pdf,
         inputs=link_state,
         outputs=[output_image, pdf_status]
+    ).then(
+    show_pdf_link,
+    inputs=link_state,
+    outputs=pdf_link_btn
     )
 
 # This block ensures the code inside only runs when the script is executed directly.