Update app.py

app.py

@@ -18,9 +18,16 @@ import random
 import logging
 import numpy as np
 
-# Logging setup
+# Logging setup with a custom buffer
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
+log_records = []  # Custom list to store log records
+
+class LogCaptureHandler(logging.Handler):
+    def emit(self, record):
+        log_records.append(record)
+
+logger.addHandler(LogCaptureHandler())
 
 # Page Configuration
 st.set_page_config(
@@ -200,6 +207,13 @@ class DiffusionBuilder:
         return self.pipeline(prompt, num_inference_steps=50).images[0]
 
 # Utility Functions
+def generate_filename(sequence, ext="png"):
+    from datetime import datetime
+    import pytz
+    central = pytz.timezone('US/Central')
+    timestamp = datetime.now(central).strftime("%d%m%Y%H%M%S%p")
+    return f"{sequence}{timestamp}.{ext}"
+
 def get_download_link(file_path, mime_type="text/plain", label="Download"):
     with open(file_path, 'rb') as f:
         data = f.read()
@@ -466,9 +480,9 @@ with tab5:
                     image = agent.generate(row["Image Idea"])
                     st.image(image, caption=f"{row['Theme']} - {row['Image Idea']}")
 
-# Main App
+# Display Logs
 st.sidebar.subheader("Action Logs 📜")
 log_container = st.sidebar.empty()
 with log_container:
-    for record in logger.handlers[0].buffer:
+    for record in log_records:
         st.write(f"{record.asctime} - {record.levelname} - {record.message}")