Deploy from GitHub main

src/ui/compi_phase3_final_dashboard.py

@@ -1367,13 +1367,23 @@ with tab_gallery:
             if Path(p).exists():
                 try:
                     df = pd.read_csv(p)
-                    df["source_log"] = Path(p).name
-                    frames.append(df)
+                    if not df.empty and "filepath" in df.columns:
+                        df["source_log"] = Path(p).name
+                        frames.append(df)
                 except Exception as e:
                     st.warning(f"Failed reading {p}: {e}")
         if not frames:
             return pd.DataFrame(columns=["filepath"])
-        return pd.concat(frames, ignore_index=True).drop_duplicates(subset=["filepath"])
+        try:
+            combined = pd.concat(frames, ignore_index=True)
+            # Extra safety: ensure filepath column exists before deduplication
+            if "filepath" in combined.columns and len(combined) > 0:
+                return combined.drop_duplicates(subset=["filepath"])
+            else:
+                return combined
+        except Exception as e:
+            st.warning(f"Failed concatenating logs: {e}")
+            return pd.DataFrame(columns=["filepath"])
 
     def scan_images():
         """Scan output directory for images"""
@@ -1397,7 +1407,24 @@ with tab_gallery:
     imgs_df = scan_images()
     logs_df = read_logs()
     ann_df = load_annotations()
-    meta_df = imgs_df.merge(logs_df, on="filepath", how="left")
+
+    # Safe merge with error handling - extra robust for Spaces
+    try:
+        if (not imgs_df.empty and not logs_df.empty and
+            "filepath" in logs_df.columns and "filepath" in imgs_df.columns and
+            len(logs_df) > 0 and len(imgs_df) > 0):
+            meta_df = imgs_df.merge(logs_df, on="filepath", how="left")
+        else:
+            meta_df = imgs_df.copy()
+            # Ensure basic columns exist
+            if "filepath" not in meta_df.columns and not meta_df.empty:
+                meta_df["filepath"] = meta_df.get("filename", "unknown")
+    except Exception as e:
+        st.warning(f"Merge failed, using image data only: {e}")
+        meta_df = imgs_df.copy()
+        # Ensure basic columns exist even after error
+        if not meta_df.empty and "filepath" not in meta_df.columns:
+            meta_df["filepath"] = meta_df.get("filename", "unknown")
 
     if meta_df.empty:
         st.info("No images found in outputs/. Generate some images first.")
@@ -1569,13 +1596,23 @@ with tab_export:
             if Path(p).exists():
                 try:
                     df = pd.read_csv(p)
-                    df["source_log"] = Path(p).name
-                    frames.append(df)
+                    if not df.empty and "filepath" in df.columns:
+                        df["source_log"] = Path(p).name
+                        frames.append(df)
                 except Exception as e:
                     st.warning(f"Read fail {p}: {e}")
         if not frames:
             return pd.DataFrame(columns=["filepath"])
-        return pd.concat(frames, ignore_index=True).drop_duplicates(subset=["filepath"])
+        try:
+            combined = pd.concat(frames, ignore_index=True)
+            # Extra safety: ensure filepath column exists before deduplication
+            if "filepath" in combined.columns and len(combined) > 0:
+                return combined.drop_duplicates(subset=["filepath"])
+            else:
+                return combined
+        except Exception as e:
+            st.warning(f"Failed concatenating export logs: {e}")
+            return pd.DataFrame(columns=["filepath"])
 
     def scan_imgs():
         """Scan images for export"""
@@ -1591,7 +1628,23 @@ with tab_export:
     if imgs_df.empty:
         st.info("No images to export yet. Generate some images first.")
     else:
-        meta_df = imgs_df.merge(logs_df, on="filepath", how="left")
+        # Safe merge with error handling - extra robust for Spaces
+        try:
+            if (not imgs_df.empty and not logs_df.empty and
+                "filepath" in logs_df.columns and "filepath" in imgs_df.columns and
+                len(logs_df) > 0 and len(imgs_df) > 0):
+                meta_df = imgs_df.merge(logs_df, on="filepath", how="left")
+            else:
+                meta_df = imgs_df.copy()
+                # Ensure basic columns exist
+                if "filepath" not in meta_df.columns and not meta_df.empty:
+                    meta_df["filepath"] = meta_df.get("filename", "unknown")
+        except Exception as e:
+            st.warning(f"Export merge failed, using image data only: {e}")
+            meta_df = imgs_df.copy()
+            # Ensure basic columns exist even after error
+            if not meta_df.empty and "filepath" not in meta_df.columns:
+                meta_df["filepath"] = meta_df.get("filename", "unknown")
 
         # Display available images
         st.markdown("### 📋 Available Images")