Update app.py

app.py

@@ -283,7 +283,7 @@ def generate_filename(sequence, ext="png"):
     import pytz
     central = pytz.timezone('US/Central')
     dt = datetime.now(central)
-    return f"{dt.strftime('%m-%d-%Y-%I-%M-%S-%p')}.{ext}"
+    return f"{dt.strftime('%m-%d-%Y-%I-%M-%S-%p')}-{sequence}.{ext}"
 
 def get_download_link(file_path, mime_type="text/plain", label="Download"):
     try:
@@ -329,31 +329,25 @@ def update_gallery():
         for idx, file in enumerate(media_files[:gallery_size * 2]):
             with cols[idx % 2]:
                 st.image(Image.open(file), caption=file, use_container_width=True)
-                st.markdown(get_download_link(file, "image/png", "Download Image"), unsafe_allow_html=True)
+                st.markdown(get_download_link(file, "image/png", "Download Snap 📸"), unsafe_allow_html=True)
 
 def get_available_video_devices():
-    # Placeholder list since we can't directly query browser devices from Python
-    # This is a fallback; ideally, use a custom component for navigator.mediaDevices.enumerateDevices()
-    video_devices = ["Camera 0", "Camera 1"]  # Default assumption
+    # Assuming 6 cameras as per your setup; adjust if needed
+    video_devices = [f"Camera {i} 🎥" for i in range(6)]
     try:
-        # Attempt OpenCV detection as a secondary check
         detected = []
         for i in range(10):
             cap = cv2.VideoCapture(i, cv2.CAP_V4L2)
             if not cap.isOpened():
-                cap = cv2.VideoCapture(i)  # Try default backend
+                cap = cv2.VideoCapture(i)
             if cap.isOpened():
-                detected.append(f"Camera {i}")
+                detected.append(f"Camera {i} 🎥")
                 logger.info(f"Detected camera at index {i}")
                 cap.release()
-            else:
-                logger.debug(f"No camera detected at index {i}")
         if detected:
-            video_devices = detected
-        else:
-            logger.warning("OpenCV detected no cameras; using browser-inferred defaults")
+            video_devices = detected[:6]  # Limit to 6 as per your spec
     except Exception as e:
-        logger.error(f"Error detecting video devices with OpenCV: {str(e)}")
+        logger.error(f"Error detecting cameras: {str(e)}")
     return video_devices
 
 def mock_search(query: str) -> str:
@@ -428,21 +422,21 @@ with col1:
         if media_files:
             zip_path = f"snapshot_collection_{int(time.time())}.zip"
             zip_files(media_files, zip_path)
-            st.sidebar.markdown(get_download_link(zip_path, "application/zip", "Download All Snapshots"), unsafe_allow_html=True)
-            st.sidebar.success("Snapshots zipped and ready! 🎉 Grab your loot!")
+            st.sidebar.markdown(get_download_link(zip_path, "application/zip", "Download All Snaps 📦"), unsafe_allow_html=True)
+            st.sidebar.success("Snapshots zipped! 🎉 Grab your loot!")
         else:
-            st.sidebar.warning("No snapshots to zip! 📸 Snap some pics first!")
+            st.sidebar.warning("No snaps to zip! 📸 Snap some first!")
 with col2:
     if st.button("Delete All 🗑️"):
         media_files = get_gallery_files(["png"])
         if media_files:
             delete_files(media_files)
-            st.sidebar.success("All snapshots vanquished! 🧹 Gallery cleared!")
+            st.sidebar.success("Snaps vanquished! 🧹 Gallery cleared!")
             update_gallery()
         else:
-            st.sidebar.warning("Nothing to delete! 📸 Snap some pics to clear later!")
+            st.sidebar.warning("Nothing to delete! 📸 Snap some pics!")
 
-uploaded_files = st.sidebar.file_uploader("Upload Files 🎵🎥🖼️📝📜", type=["mp3", "mp4", "png", "jpeg", "md", "pdf", "docx"], accept_multiple_files=True)
+uploaded_files = st.sidebar.file_uploader("Upload Goodies 🎵🎥🖼️📝📜", type=["mp3", "mp4", "png", "jpeg", "md", "pdf", "docx"], accept_multiple_files=True)
 if uploaded_files:
     for uploaded_file in uploaded_files:
         filename = uploaded_file.name
@@ -455,14 +449,14 @@ audio_files = get_gallery_files(["mp3"])
 if audio_files:
     for file in audio_files[:gallery_size]:
         st.sidebar.audio(file, format="audio/mp3")
-        st.sidebar.markdown(get_download_link(file, "audio/mp3", f"Download {file}"), unsafe_allow_html=True)
+        st.sidebar.markdown(get_download_link(file, "audio/mp3", f"Grab Tune 🎵"), unsafe_allow_html=True)
 
 st.sidebar.subheader("Video Gallery 🎥")
 video_files = get_gallery_files(["mp4"])
 if video_files:
     for file in video_files[:gallery_size]:
         st.sidebar.video(file, format="video/mp4")
-        st.sidebar.markdown(get_download_link(file, "video/mp4", f"Download {file}"), unsafe_allow_html=True)
+        st.sidebar.markdown(get_download_link(file, "video/mp4", f"Snag Clip 🎬"), unsafe_allow_html=True)
 
 st.sidebar.subheader("Image Gallery 🖼️")
 image_files = get_gallery_files(["png", "jpeg"])
@@ -471,7 +465,7 @@ if image_files:
     for idx, file in enumerate(image_files[:gallery_size * 2]):
         with cols[idx % 2]:
             st.image(Image.open(file), caption=file, use_container_width=True)
-            st.markdown(get_download_link(file, "image/png" if file.endswith(".png") else "image/jpeg", f"Download {file}"), unsafe_allow_html=True)
+            st.markdown(get_download_link(file, "image/png" if file.endswith(".png") else "image/jpeg", f"Save Pic 🖼️"), unsafe_allow_html=True)
 
 st.sidebar.subheader("Markdown Gallery 📝")
 md_files = get_gallery_files(["md"])
@@ -479,14 +473,14 @@ if md_files:
     for file in md_files[:gallery_size]:
         with open(file, "r") as f:
             st.sidebar.markdown(f.read())
-        st.sidebar.markdown(get_download_link(file, "text/markdown", f"Download {file}"), unsafe_allow_html=True)
+        st.sidebar.markdown(get_download_link(file, "text/markdown", f"Get Note 📝"), unsafe_allow_html=True)
 
 st.sidebar.subheader("Document Gallery 📜")
 doc_files = get_gallery_files(["pdf", "docx"])
 if doc_files:
     for file in doc_files[:gallery_size]:
         mime_type = "application/pdf" if file.endswith(".pdf") else "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-        st.sidebar.markdown(get_download_link(file, mime_type, f"Download {file}"), unsafe_allow_html=True)
+        st.sidebar.markdown(get_download_link(file, mime_type, f"Fetch Doc 📜"), unsafe_allow_html=True)
 
 st.sidebar.subheader("Model Management 🗂️")
 model_type = st.sidebar.selectbox("Model Type", ["Causal LM", "Diffusion"])
@@ -549,46 +543,73 @@ with tab1:
             st.error(f"Model build failed: {str(e)} 💥 (Check logs for details!)")
 
 with tab2:
-    st.header("Camera Snap 📷 (Dual Capture!)")
+    st.header("Camera Snap 📷 (Dual Capture Fiesta!)")
     video_devices = get_available_video_devices()
-    st.subheader("Camera Settings ⚙️")
-    st.write(f"Available Cameras: {', '.join(video_devices)}")
-    st.info("Camera selection is informational. Use your browser settings to switch cameras (e.g., Chrome > Settings > Privacy > Camera).")
-
-    # Camera selection with defaults
-    default_cam0_index = 0
-    default_cam1_index = 1 if len(video_devices) > 1 else 0
-    selected_cam0 = st.selectbox("Select Camera 0", video_devices, index=default_cam0_index, key="cam0_select")
-    selected_cam1 = st.selectbox("Select Camera 1", video_devices, index=default_cam1_index, key="cam1_select")
-    
-    slice_count = st.number_input("Image Slice Count 🎞️", min_value=1, max_value=20, value=10, help="How many snaps to dream of? (Automation’s on vacation! 😜)")
-    video_length = st.number_input("Video Dream Length (seconds) 🎥", min_value=1, max_value=30, value=10, help="Imagine a vid this long—sadly, we’re stuck with pics for now! 😂")
-    
+    st.write(f"🎉 Detected Cameras: {', '.join(video_devices)}")
+    st.info("Switch cams in your browser settings (e.g., Chrome > Privacy > Camera) since I’m a browser star! 🌟")
+
+    # Camera 0 Settings
+    st.subheader("Camera 0 🎬 - Lights, Camera, Action!")
+    cam0_cols = st.columns(4)
+    with cam0_cols[0]:
+        cam0_device = st.selectbox("Cam 📷", video_devices, index=0, key="cam0_device", help="Pick your star cam! 🌟")
+    with cam0_cols[1]:
+        cam0_label = st.text_input("Tag 🏷️", "Cam 0 Snap", key="cam0_label", help="Name your masterpiece! 🎨")
+    with cam0_cols[2]:
+        cam0_help = st.text_input("Hint 💡", "Snap a heroic moment! 🦸‍♂️", key="cam0_help", help="Give a fun tip!")
+    with cam0_cols[3]:
+        cam0_vis = st.selectbox("Show 🖼️", ["visible", "hidden", "collapsed"], index=0, key="cam0_vis", help="Label vibes: Visible, Sneaky, or Gone!")
+
+    # Camera 1 Settings
+    st.subheader("Camera 1 🎥 - Roll the Film!")
+    cam1_cols = st.columns(4)
+    with cam1_cols[0]:
+        cam1_device = st.selectbox("Cam 📷", video_devices, index=1 if len(video_devices) > 1 else 0, key="cam1_device", help="Choose your blockbuster cam! 🎬")
+    with cam1_cols[1]:
+        cam1_label = st.text_input("Tag 🏷️", "Cam 1 Snap", key="cam1_label", help="Title your epic shot! 🌠")
+    with cam1_cols[2]:
+        cam1_help = st.text_input("Hint 💡", "Grab an epic frame! 🌟", key="cam1_help", help="Drop a cheeky hint!")
+    with cam1_cols[3]:
+        cam1_vis = st.selectbox("Show 🖼️", ["visible", "hidden", "collapsed"], index=0, key="cam1_vis", help="Label style: Show it, Hide it, Poof!")
+
+    # Capture Widgets
     cols = st.columns(2)
     with cols[0]:
-        st.subheader(f"Camera 0 ({selected_cam0}) 🎬")
-        cam0_img = st.camera_input("Snap a Shot - Cam 0 📸", key="cam0", help="Click to capture a heroic moment! 🦸‍♂️")
+        st.subheader(f"Camera 0 ({cam0_device}) 🎬")
+        cam0_img = st.camera_input(
+            label=cam0_label,
+            key="cam0",
+            help=cam0_help,
+            disabled=False,
+            label_visibility=cam0_vis
+        )
         if cam0_img:
-            filename = generate_filename(0)
+            filename = generate_filename("cam0")
             with open(filename, "wb") as f:
                 f.write(cam0_img.getvalue())
             st.image(Image.open(filename), caption=filename, use_container_width=True)
             logger.info(f"Saved snapshot from Camera 0: {filename}")
             st.session_state['captured_images'].append(filename)
             update_gallery()
-        st.info("🚨 Multi-frame capture’s on strike! Snap one at a time—your Titan’s too cool for automation glitches! 😎")
+        st.info("🚨 One snap at a time—your Titan’s too cool for bursts! 😎")
     with cols[1]:
-        st.subheader(f"Camera 1 ({selected_cam1}) 🎥")
-        cam1_img = st.camera_input("Snap a Shot - Cam 1 📸", key="cam1", help="Grab another epic frame! 🌟")
+        st.subheader(f"Camera 1 ({cam1_device}) 🎥")
+        cam1_img = st.camera_input(
+            label=cam1_label,
+            key="cam1",
+            help=cam1_help,
+            disabled=False,
+            label_visibility=cam1_vis
+        )
         if cam1_img:
-            filename = generate_filename(1)
+            filename = generate_filename("cam1")
             with open(filename, "wb") as f:
                 f.write(cam1_img.getvalue())
             st.image(Image.open(filename), caption=filename, use_container_width=True)
             logger.info(f"Saved snapshot from Camera 1: {filename}")
             st.session_state['captured_images'].append(filename)
             update_gallery()
-        st.info("🚨 Frame bursts? Nope, manual snaps only! One click, one masterpiece! 🎨")
+        st.info("🚨 Single shots only—craft your masterpiece! 🎨")
 
 with tab3:
     st.header("Fine-Tune Titan (NLP) 🔧 (Teach Your Word Wizard Some Tricks!)")
@@ -606,7 +627,7 @@ with tab3:
                 writer = csv.DictWriter(f, fieldnames=["prompt", "response"])
                 writer.writeheader()
                 writer.writerows(sample_data)
-            st.markdown(get_download_link(csv_path, "text/csv", "Download Sample CSV"), unsafe_allow_html=True)
+            st.markdown(get_download_link(csv_path, "text/csv", "Download Sample CSV 📜"), unsafe_allow_html=True)
             st.success(f"Sample CSV generated as {csv_path}! ✅ (Fresh from the data oven!)")
         uploaded_csv = st.file_uploader("Upload CSV for SFT 📜", type="csv", help="Feed your Titan some tasty prompt-response pairs! 🍽️")
         if uploaded_csv and st.button("Fine-Tune with Uploaded CSV 🔄"):
@@ -622,7 +643,7 @@ with tab3:
                 status.update(label="Fine-tuning completed! 🎉 (Wordsmith Titan unleashed!)", state="complete")
             zip_path = f"{new_config.model_path}.zip"
             zip_files([new_config.model_path], zip_path)
-            st.markdown(get_download_link(zip_path, "application/zip", "Download Fine-Tuned NLP Titan"), unsafe_allow_html=True)
+            st.markdown(get_download_link(zip_path, "application/zip", "Download Fine-Tuned NLP Titan 📦"), unsafe_allow_html=True)
 
 with tab4:
     st.header("Test Titan (NLP) 🧪 (Put Your Word Wizard to the Test!)")
@@ -721,14 +742,14 @@ with tab6:
                     status.update(label="Fine-tuning completed! 🎉 (Pixel Titan unleashed!)", state="complete")
                 zip_path = f"{new_config.model_path}.zip"
                 zip_files([new_config.model_path], zip_path)
-                st.markdown(get_download_link(zip_path, "application/zip", "Download Fine-Tuned CV Titan"), unsafe_allow_html=True)
+                st.markdown(get_download_link(zip_path, "application/zip", "Download Fine-Tuned CV Titan 📦"), unsafe_allow_html=True)
             csv_path = f"sft_dataset_{int(time.time())}.csv"
             with open(csv_path, "w", newline="") as f:
                 writer = csv.writer(f)
                 writer.writerow(["image", "text"])
                 for _, row in edited_data.iterrows():
                     writer.writerow([row["image"], row["text"]])
-            st.markdown(get_download_link(csv_path, "text/csv", "Download SFT Dataset CSV"), unsafe_allow_html=True)
+            st.markdown(get_download_link(csv_path, "text/csv", "Download SFT Dataset CSV 📜"), unsafe_allow_html=True)
 
 with tab7:
     st.header("Test Titan (CV) 🧪 (Unleash Your Pixel Power!)")