Update app.py

app.py

@@ -55,14 +55,14 @@ st.set_page_config(
 # Initialize st.session_state
 if 'captured_files' not in st.session_state:
     st.session_state['captured_files'] = {'cam0': None, 'cam1': None}  # One file per camera
+if 'history' not in st.session_state:
+    st.session_state['history'] = {'cam0': None, 'cam1': None}  # One history entry per camera
 if 'builder' not in st.session_state:
     st.session_state['builder'] = None
 if 'model_loaded' not in st.session_state:
     st.session_state['model_loaded'] = False
 if 'processing' not in st.session_state:
     st.session_state['processing'] = {}
-if 'history' not in st.session_state:
-    st.session_state['history'] = {'cam0': None, 'cam1': None}  # One history entry per camera
 
 # Model Configuration Classes
 @dataclass
@@ -311,7 +311,7 @@ class DiffusionBuilder:
 # Utility Functions
 def generate_filename(sequence, ext="png"):
     timestamp = time.strftime("%d%m%Y%H%M%S")
-    return f"{sequence}{timestamp}.{ext}"
+    return f"{sequence}_{timestamp}.{ext}"
 
 def get_download_link(file_path, mime_type="text/plain", label="Download"):
     with open(file_path, 'rb') as f:
@@ -512,9 +512,12 @@ def update_gallery():
     valid_files = [f for f in media_files if f and os.path.exists(f)]  # Only valid files
     if valid_files:
         cols = st.sidebar.columns(2)
-        for idx, file in enumerate(valid_files[:2]):  # Limit to 2 images
-            with cols[idx]:
-                st.image(Image.open(file), caption=os.path.basename(file), use_container_width=True)
+        if st.session_state['captured_files']['cam0'] in valid_files:
+            with cols[0]:
+                st.image(Image.open(st.session_state['captured_files']['cam0']), caption="Camera 0", use_container_width=True)
+        if st.session_state['captured_files']['cam1'] in valid_files:
+            with cols[1]:
+                st.image(Image.open(st.session_state['captured_files']['cam1']), caption="Camera 1", use_container_width=True)
 update_gallery()
 
 st.sidebar.subheader("Model Management 🗂️")
@@ -539,7 +542,7 @@ st.sidebar.subheader("History 📜")
 history_container = st.sidebar.empty()
 with history_container:
     valid_history = [st.session_state['history']['cam0'], st.session_state['history']['cam1']]
-    for entry in [e for e in valid_history if e][:2]:  # Show only latest two
+    for entry in [e for e in valid_history if e]:  # Show only non-None entries
         st.write(entry)
 
 # Tabs