Update

app.py

@@ -7,11 +7,15 @@ import pandas as pd
 from PIL import Image
 from tqdm import tqdm
 import logging
+import json
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Check for environment variables
+GOOGLE_CREDENTIALS = os.getenv('GOOGLE_CREDENTIALS')
+
 class DatasetManager:
     def __init__(self, local_images_dir="downloaded_cards"):
         self.local_images_dir = local_images_dir
@@ -24,10 +28,31 @@ class DatasetManager:
     def authenticate_drive(self):
         """Authenticate with Google Drive"""
         try:
+            # Create client_secrets.json from environment variable if available
+            if GOOGLE_CREDENTIALS:
+                with open('client_secrets.json', 'w') as f:
+                    f.write(GOOGLE_CREDENTIALS)
+                
             gauth = GoogleAuth()
-            # Specify the path to client_secrets.json
-            gauth.settings['client_config_file'] = 'client_secrets.json'  # Make sure this matches your file path
-            gauth.LocalWebserverAuth()
+            # Configure for headless authentication
+            gauth.settings['get_refresh_token'] = True
+            gauth.LoadCredentialsFile("mycreds.txt")
+            
+            if gauth.credentials is None:
+                # Create local webserver for authentication if running locally
+                if os.getenv('SPACE_ID') is None:
+                    gauth.LocalWebserverAuth()
+                else:
+                    # For Hugging Face Spaces, use service account or saved credentials
+                    gauth.CommandLineAuth()
+            elif gauth.access_token_expired:
+                gauth.Refresh()
+            else:
+                gauth.Authorize()
+            
+            # Save the credentials for future use
+            gauth.SaveCredentialsFile("mycreds.txt")
+            
             self.drive = GoogleDrive(gauth)
             return True, "Successfully authenticated with Google Drive"
         except Exception as e:
@@ -131,25 +156,52 @@ def process_pipeline(folder_id, naming_convention):
     success, hf_message = manager.update_huggingface_dataset(renamed_files)
     return f"{message}\n{hf_message}"
 
+# Authentication status interface
+def check_auth_status():
+    try:
+        gauth = GoogleAuth()
+        gauth.LoadCredentialsFile("mycreds.txt")
+        if gauth.credentials is not None and not gauth.access_token_expired:
+            return "🟢 Authenticated with Google Drive"
+        else:
+            return "🔴 Not authenticated with Google Drive"
+    except Exception as e:
+        return f"🔴 Error checking authentication: {str(e)}"
+
 # Gradio interface
-demo = gr.Interface(
-    fn=process_pipeline,
-    inputs=[
-        gr.Textbox(
+with gr.Blocks() as demo:
+    gr.Markdown("# Sports Cards Dataset Processor")
+    gr.Markdown("Download card images from Google Drive and add them to the sports-cards dataset")
+    
+    # Authentication status
+    auth_status = gr.Textbox(
+        label="Authentication Status",
+        value=check_auth_status(),
+        interactive=False
+    )
+    
+    with gr.Row():
+        folder_id = gr.Textbox(
             label="Google Drive File/Folder ID",
             placeholder="Enter the ID from your Google Drive URL",
             value="151VOxPO91mg0C3ORiioGUd4hogzP1ujm"
-        ),
-        gr.Textbox(
+        )
+        naming = gr.Textbox(
             label="Naming Convention",
             placeholder="e.g., sports_card",
             value="sports_card"
         )
-    ],
-    outputs=gr.Textbox(label="Status"),
-    title="Sports Cards Dataset Processor",
-    description="Download card images from Google Drive and add them to the sports-cards dataset"
-)
+    
+    # Process button and output
+    process_btn = gr.Button("Process Images")
+    output = gr.Textbox(label="Status")
+    
+    # Handle processing
+    process_btn.click(
+        fn=process_pipeline,
+        inputs=[folder_id, naming],
+        outputs=output
+    )
 
 if __name__ == "__main__":
     demo.launch()