fix

app.py

@@ -11,6 +11,7 @@ import base64
 import logging
 import sys
 import shutil
+import zipfile
 
 # Configure logging
 # Simplified logging for cloud deployment
@@ -281,6 +282,18 @@ def process_business_cards(images, model_name="gemini-2.5-flash", save_images=Tr
         
         logger.info(f"Successfully saved {len(saved_image_paths)} images")
         
+        # Create a ZIP file of saved images
+        if len(saved_image_paths) > 0:
+            zip_filename = f"business_cards_{timestamp}.zip"
+            zip_path = Path(zip_filename)
+            try:
+                with zipfile.ZipFile(zip_path, 'w') as zipf:
+                    for img_path in saved_image_paths:
+                        zipf.write(img_path, os.path.basename(img_path))
+                logger.info(f"Created ZIP file with {len(saved_image_paths)} images: {zip_filename}")
+            except Exception as e:
+                logger.error(f"Failed to create ZIP file: {e}")
+        
         # List directory contents for debugging
         try:
             images_list = list(images_dir.iterdir())
@@ -449,6 +462,14 @@ def process_business_cards(images, model_name="gemini-2.5-flash", save_images=Tr
             logger.debug(f"Adjusted column widths: {adjusted_columns}")
         
         logger.info(f"Current run Excel file saved successfully: {current_filename}")
+        
+        # Also save a copy in root directory for Hugging Face Spaces visibility
+        root_current_file = Path(f"current_run_{timestamp}.xlsx")
+        try:
+            shutil.copy2(current_filename, root_current_file)
+            logger.info(f"Copied current run file to root: {root_current_file}")
+        except Exception as e:
+            logger.error(f"Failed to copy current file to root: {e}")
     except Exception as e:
         logger.error(f"Failed to write current run Excel file: {e}")
         raise
@@ -475,6 +496,14 @@ def process_business_cards(images, model_name="gemini-2.5-flash", save_images=Tr
             logger.debug(f"Adjusted column widths: {adjusted_columns}")
         
         logger.info(f"Cumulative Excel file saved successfully: {cumulative_filename}")
+        
+        # Also save a copy in root directory for Hugging Face Spaces visibility
+        root_cumulative_file = Path("all_business_cards_total.xlsx")
+        try:
+            shutil.copy2(cumulative_filename, root_cumulative_file)
+            logger.info(f"Copied cumulative file to root: {root_cumulative_file}")
+        except Exception as e:
+            logger.error(f"Failed to copy cumulative file to root: {e}")
     except Exception as e:
         logger.error(f"Failed to write cumulative Excel file: {e}")
         raise
@@ -496,6 +525,14 @@ def process_business_cards(images, model_name="gemini-2.5-flash", save_images=Tr
     summary += f"📁 Total cumulative file: {cumulative_filename.name}\n"
     summary += f"📊 Total cards in database: {len(cumulative_df)}\n\n"
     
+    # Add note about root directory files for HF Spaces
+    summary += "📌 Note: Files are also saved in the root directory for easy access in Hugging Face Spaces Files tab.\n"
+    summary += "   - current_run_[timestamp].xlsx\n"
+    summary += "   - all_business_cards_total.xlsx\n"
+    if save_images and 'saved_image_paths' in locals() and len(saved_image_paths) > 0:
+        summary += f"   - business_cards_{timestamp}.zip (contains all images)\n"
+    summary += "\n"
+    
     if errors:
         logger.warning(f"Encountered {len(errors)} errors during processing")
         summary += "Errors encountered:\n" + "\n".join(errors)
@@ -532,6 +569,15 @@ def process_business_cards(images, model_name="gemini-2.5-flash", save_images=Tr
     except Exception as e:
         logger.error(f"Error listing export directory: {e}")
     
+    # List root directory Excel/ZIP files for debugging
+    try:
+        root_files = [f for f in Path(".").iterdir() if f.suffix in ['.xlsx', '.zip'] and f.is_file()]
+        logger.info(f"Root directory contains {len(root_files)} Excel/ZIP files")
+        for root_file in root_files[-5:]:  # Show last 5 files
+            logger.info(f"  - {root_file.name} ({root_file.stat().st_size} bytes)")
+    except Exception as e:
+        logger.error(f"Error listing root directory: {e}")
+    
     # Return string paths for Gradio File components
     return str(current_filename), str(cumulative_filename), summary, preview_df
 
@@ -552,6 +598,11 @@ with gr.Blocks(title="Business Card Data Extractor") as demo:
         **Image Storage:**
         - 💾 **Optional**: Save uploaded images to business_cards folder
         - 📝 **Tracking**: Image file paths included in Excel database
+        
+        **📌 Finding Your Files in Hugging Face Spaces:**
+        - Click the "Files" tab at the top of this page
+        - Look for: `current_run_*.xlsx`, `all_business_cards_total.xlsx`, and `business_cards_*.zip`
+        - These files are copied to the root directory for easy access
         """
     )