Update app.py

app.py

@@ -519,11 +519,46 @@ with gr.Blocks(title="🤖 Dox the Data Professional's Guide 🤖"#,
                         elem_classes=["primary-action"]
                     )
 
-                    clear_btn = gr.Button(
-                        "🧹 Clear Chat",
-                        variant="secondary",
-                        elem_classes=["clear-action"]
-                    )
+                with gr.Row():
+                    ask_btn = gr.Button("Submit 📤", variant="primary", elem_classes="primary-action")
+                    clear_btn = gr.Button("🧹 Clear Chat", elem_classes="clear-action")
+                # A section for example questions.
+                gr.Markdown("### 💡 Example Questions", elem_classes="component")
+                examples = gr.Examples(
+                    examples=example_questions
+                    inputs=question,
+                    outputs=[chatbot, question],
+                    fn=run_example,
+                    cache_examples=False
+                )
+
+                # 👍👎 Feedback buttons
+                with gr.Row():
+                    thumbs_up = gr.Button("👍 Helpful", elem_classes="component")
+                    thumbs_down = gr.Button("👎 Not Helpful", elem_classes="component")
+                 
+                # 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, elem_classes="component")
+                feedback_status = gr.Markdown("", elem_classes="component")
+            # RIGHT-SIDE COLUMN: for the PDF preview.
+            with gr.Column(scale=3):
+                gr.Markdown("### 📄 Referenced PDF Document (🌐 Empty for Web Results)", elem_classes="component")
+                #gr.Markdown(" 🌐 Empty by default", elem_classes="component")
+                # A hidden state to store the PDF link found in the agent's response.
+                link_state = gr.State()
+                # A markdown component to show PDF loading status.
+                pdf_status = gr.Markdown(visible=False, elem_classes="component")
+                # An image component to display the PDF preview.
+                output_image = gr.Image(
+                    label="⬇️ Cheat Sheet Preview",
+                    visible=False
+                )
+                pdf_link_btn = gr.Markdown("")
     
     # # Add titles and descriptions using Markdown.
     # gr.Markdown("# 🤖 Dox the Data Professional's Advisor 🤖")
@@ -593,153 +628,153 @@ with gr.Blocks(title="🤖 Dox the Data Professional's Guide 🤖"#,
     #         )
     #         pdf_link_btn = gr.Markdown("")
 
-    # # Defines the main chat logic as a generator function for streaming output.
-    # def chat_ui(user_message, chat_history):
-    #     # Initialize chat history if it's the first turn.
-    #     if chat_history is None:
-    #         chat_history = []
+    # Defines the main chat logic as a generator function for streaming output.
+    def chat_ui(user_message, chat_history):
+        # Initialize chat history if it's the first turn.
+        if chat_history is None:
+            chat_history = []
     
-    #     # Append the user's message to the chat history.
-    #     chat_history.append({
-    #         "role": "user",
-    #         "content": user_message
-    #     })
+        # Append the user's message to the chat history.
+        chat_history.append({
+            "role": "user",
+            "content": user_message
+        })
     
-    #     # Append a temporary "Thinking..." message from the assistant.
-    #     chat_history.append({
-    #         "role": "assistant",
-    #         "content": "🤔 Dox is thinking..."
-    #     })
+        # Append a temporary "Thinking..." message from the assistant.
+        chat_history.append({
+            "role": "assistant",
+            "content": "🤔 Dox is thinking..."
+        })
     
-    #     # `yield` immediately updates the UI with the user's message and "Thinking...".
-    #     # It also clears the user's input textbox.
-    #     yield (
-    #         chat_history,
-    #         None, # No link yet.
-    #         gr.update(value=None, visible=False), # Hide image preview.
-    #         ""  # Clear textbox.
-    #     )
+        # `yield` immediately updates the UI with the user's message and "Thinking...".
+        # It also clears the user's input textbox.
+        yield (
+            chat_history,
+            None, # No link yet.
+            gr.update(value=None, visible=False), # Hide image preview.
+            ""  # Clear textbox.
+        )
     
-    #     # Call the agent to get the actual response and PDF link.
-    #     response_text, link = ask_agent(user_message)
+        # Call the agent to get the actual response and PDF link.
+        response_text, link = ask_agent(user_message)
     
-    #     # Replace the "Thinking..." message with the final response from the agent.
-    #     chat_history[-1] = {
-    #         "role": "assistant",
-    #         "content": response_text
-    #     }
+        # Replace the "Thinking..." message with the final response from the agent.
+        chat_history[-1] = {
+            "role": "assistant",
+            "content": response_text
+        }
     
-    #     # `yield` again to update the UI with the final response.
-    #     yield (
-    #         chat_history,
-    #         link, # Pass the extracted link to the link_state.
-    #         gr.update(value=None, visible=False), # Keep image preview hidden for now.
-    #         "" # Keep textbox clear.
-    #     )
+        # `yield` again to update the UI with the final response.
+        yield (
+            chat_history,
+            link, # Pass the extracted link to the link_state.
+            gr.update(value=None, visible=False), # Keep image preview hidden for now.
+            "" # Keep textbox clear.
+        )
         
-    # # This is a helper function to avoid repeating the event handler chain.
-    # def submit_chain():
-    #     # It specifies that `chat_ui` is the function to run.
-    #     # It maps the `question` textbox and `chatbot` history as inputs.
-    #     # It maps the outputs to `chatbot` history, `link_state`, `output_image`, and clears the `question` textbox.
-    #     return (
-    #         chat_ui,
-    #         [question, chatbot],
-    #         [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), gr.update(value=None, visible=False), gr.update(value=None, visible=False)
-
-    # clear_btn.click(
-    #     clear_chat,
-    #     outputs=[chatbot, link_state, output_image, feedback_box, submit_feedback_btn]
-    # )
-
-    # def show_feedback_box():
-    #     return gr.update(visible=True), gr.update(visible=True)
-
-    # def show_appreciation():
-    #     logger.info("It was helpful!")
-    #     return "✅ Feedback submitted. Thank you!"
+    # This is a helper function to avoid repeating the event handler chain.
+    def submit_chain():
+        # It specifies that `chat_ui` is the function to run.
+        # It maps the `question` textbox and `chatbot` history as inputs.
+        # It maps the outputs to `chatbot` history, `link_state`, `output_image`, and clears the `question` textbox.
+        return (
+            chat_ui,
+            [question, chatbot],
+            [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), gr.update(value=None, visible=False), gr.update(value=None, visible=False)
+
+    clear_btn.click(
+        clear_chat,
+        outputs=[chatbot, link_state, output_image, feedback_box, submit_feedback_btn]
+    )
+
+    def show_feedback_box():
+        return gr.update(visible=True), gr.update(visible=True)
+
+    def show_appreciation():
+        logger.info("It was helpful!")
+        return "✅ Feedback submitted. Thank you!"
      
-    # thumbs_down.click(
-    #     show_feedback_box,
-    #     outputs=[feedback_box, submit_feedback_btn]
-    # )
-
-    # thumbs_up.click(
-    #     show_appreciation,
-    #     outputs=feedback_status
-    # )
-
-    # def handle_feedback(text):
-    #     logger.info(f"User feedback: {text}")
-    #     return "✅ Feedback submitted. Thank you!"
+    thumbs_down.click(
+        show_feedback_box,
+        outputs=[feedback_box, submit_feedback_btn]
+    )
+
+    thumbs_up.click(
+        show_appreciation,
+        outputs=feedback_status
+    )
+
+    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
-    # )
-
-    # examples.dataset.click(
-    #     *submit_chain()
-    # ).then(
-    #     prepare_pdf_loading,
-    #     inputs=link_state,
-    #     outputs=pdf_status
-    # ).then(
-    #     display_pdf,
-    #     inputs=link_state,
-    #     outputs=[output_image, pdf_status]
-    # ).then(
-    #     show_pdf_link,
-    #     inputs=link_state,
-    #     outputs=pdf_link_btn
-    # )
-
-    # # Set up the event handler for the "Submit" button click.
-    # ask_btn.click(
-    #     *submit_chain()
-    # # `.then()` chains subsequent actions after the first one completes.
-    # ).then(
-    #     # After chat_ui, call `prepare_pdf_loading` to show the "loading" message.
-    #     prepare_pdf_loading,
-    #     inputs=link_state,  # Use the link from chat_ui's output.
-    #     outputs=pdf_status # Update the pdf_status text.
-    # ).then(
-    #     # Finally, call `display_pdf` to render the PDF page.
-    #     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
-    # )
+    submit_feedback_btn.click(
+        handle_feedback,
+        inputs=feedback_box,
+        outputs=feedback_status
+    )
+
+    examples.dataset.click(
+        *submit_chain()
+    ).then(
+        prepare_pdf_loading,
+        inputs=link_state,
+        outputs=pdf_status
+    ).then(
+        display_pdf,
+        inputs=link_state,
+        outputs=[output_image, pdf_status]
+    ).then(
+        show_pdf_link,
+        inputs=link_state,
+        outputs=pdf_link_btn
+    )
+
+    # Set up the event handler for the "Submit" button click.
+    ask_btn.click(
+        *submit_chain()
+    # `.then()` chains subsequent actions after the first one completes.
+    ).then(
+        # After chat_ui, call `prepare_pdf_loading` to show the "loading" message.
+        prepare_pdf_loading,
+        inputs=link_state,  # Use the link from chat_ui's output.
+        outputs=pdf_status # Update the pdf_status text.
+    ).then(
+        # Finally, call `display_pdf` to render the PDF page.
+        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.
-    # question.submit(
-    #     *submit_chain()
-    # ).then(
-    #     prepare_pdf_loading,
-    #     inputs=link_state,
-    #     outputs=pdf_status
-    # ).then(
-    #     display_pdf,
-    #     inputs=link_state,
-    #     outputs=[output_image, pdf_status]
-    # ).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.
+    question.submit(
+        *submit_chain()
+    ).then(
+        prepare_pdf_loading,
+        inputs=link_state,
+        outputs=pdf_status
+    ).then(
+        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.
 if __name__ == "__main__":