update

app.py

@@ -7,14 +7,15 @@ import pandas as pd
 from PIL import Image
 from tqdm import tqdm
 import logging
-import json
+import yaml
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Check for environment variables
-GOOGLE_CREDENTIALS = os.getenv('GOOGLE_CREDENTIALS')
+# Load settings
+with open('settings.yaml', 'r') as file:
+    settings = yaml.safe_load(file)
 
 class DatasetManager:
     def __init__(self, local_images_dir="downloaded_cards"):
@@ -28,30 +29,25 @@ 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()
-            # Configure for headless authentication
-            gauth.settings['get_refresh_token'] = True
-            gauth.LoadCredentialsFile("mycreds.txt")
+            # Use the settings from yaml file
+            gauth.settings['client_config_file'] = settings['client_secrets_file']
+            
+            # Try to load saved credentials
+            gauth.LoadCredentialsFile("credentials.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()
+                # Authenticate if no credentials found
+                gauth.LocalWebserverAuth()
             elif gauth.access_token_expired:
+                # Refresh them if expired
                 gauth.Refresh()
             else:
+                # Initialize the saved credentials
                 gauth.Authorize()
-            
+                
             # Save the credentials for future use
-            gauth.SaveCredentialsFile("mycreds.txt")
+            gauth.SaveCredentialsFile("credentials.txt")
             
             self.drive = GoogleDrive(gauth)
             return True, "Successfully authenticated with Google Drive"
@@ -156,52 +152,25 @@ 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
-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(
+demo = gr.Interface(
+    fn=process_pipeline,
+    inputs=[
+        gr.Textbox(
             label="Google Drive File/Folder ID",
             placeholder="Enter the ID from your Google Drive URL",
             value="151VOxPO91mg0C3ORiioGUd4hogzP1ujm"
-        )
-        naming = gr.Textbox(
+        ),
+        gr.Textbox(
             label="Naming Convention",
             placeholder="e.g., sports_card",
             value="sports_card"
         )
-    
-    # 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
-    )
+    ],
+    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"
+)
 
 if __name__ == "__main__":
     demo.launch()