Update app.py

app.py

@@ -24,6 +24,26 @@ import plotly.express as px
 import pydeck as pdk
 
 
+import logging
+import streamlit as st
+from transformers import load_tool, Agent
+
+# Configure the logging settings for transformers
+transformers_logger = logging.getLogger("transformers.file_utils")
+transformers_logger.setLevel(logging.INFO)  # Set the desired logging level
+  
+
+
+ 
+
+
+
+
+
+
+
+
+
 import time  
 from transformers import load_tool, Agent
 import torch
@@ -292,6 +312,36 @@ with tabs[3]:
         - Ensure proper configuration, such as setting the Hugging Face token as an environment variable.
 
       ''')
+
+
+        # Display logs in the frontend
+    logs_expander = st.expander("Logs")
+    with logs_expander:
+        log_output = st.empty()
+    
+        # Custom logging handler to append log messages to the chat
+        class ChatHandler(logging.Handler):
+            def __init__(self):
+                super().__init__()
+    
+            def emit(self, record):
+                log_message = self.format(record)
+                with st.chat_message("ai"):
+                    st.markdown(f"Log: {log_message}")
+    
+        # Add the custom handler to the transformers_logger
+        chat_handler = ChatHandler()
+        transformers_logger.addHandler(chat_handler)
+    
+        # Function to update logs in the frontend
+        def update_logs():
+            log_output.code("")  # Clear previous logs
+            # Do nothing here since logs are appended to the chat
+    
+        # Update logs when the button is clicked
+        if st.button("Update Logs"):
+            update_logs()
+    
     
 # Chat code (user input, agent responses, etc.)
 if "messages" not in st.session_state: