Update app.py

app.py

@@ -541,153 +541,153 @@ with gr.Blocks(
     #         )
     #         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__":