Update app.py

app.py

@@ -87,26 +87,77 @@ def setup_google_drive():
         print(f"Google Drive setup failed: {e}")
         return None
 
+# Google Drive Setup with DEBUGGING
+def setup_google_drive():
+    """Initialize Google Drive service with detailed debugging"""
+    try:
+        print("Setting up Google Drive...")
+        
+        # Get service account credentials from environment variable
+        credentials_json = os.getenv('GOOGLE_SERVICE_ACCOUNT_JSON')
+        if not credentials_json:
+            print("❌ ERROR: GOOGLE_SERVICE_ACCOUNT_JSON environment variable not found")
+            return None
+            
+        print("✅ Found Google service account JSON")
+        
+        # Get Shared Drive ID
+        SHARED_DRIVE_ID = os.getenv('SHARED_DRIVE_ID')
+        if not SHARED_DRIVE_ID:
+            print("❌ ERROR: SHARED_DRIVE_ID environment variable not set")
+            return None
+            
+        print(f"✅ Shared Drive ID: {SHARED_DRIVE_ID}")
+        
+        # Parse the JSON credentials
+        try:
+            service_account_info = json.loads(credentials_json)
+            client_email = service_account_info.get('client_email', 'Unknown')
+            print(f"✅ Service account email: {client_email}")
+        except json.JSONDecodeError as e:
+            print(f"❌ ERROR: Invalid JSON in service account credentials: {e}")
+            return None
+        
+        credentials = service_account.Credentials.from_service_account_info(
+            service_account_info,
+            scopes=['https://www.googleapis.com/auth/drive.file']
+        )
+        
+        # Build the Drive service
+        drive_service = build('drive', 'v3', credentials=credentials)
+        print("✅ Google Drive service initialized successfully")
+        return drive_service
+        
+    except Exception as e:
+        print(f"❌ Google Drive setup failed: {str(e)}")
+        import traceback
+        traceback.print_exc()
+        return None
+
 # Initialize Google Drive service
 drive_service = setup_google_drive()
-
-# Get Shared Drive ID from environment variable
 SHARED_DRIVE_ID = os.getenv('SHARED_DRIVE_ID')
-if not SHARED_DRIVE_ID:
-    print("WARNING: SHARED_DRIVE_ID not set. Google Drive saving will fail.")
 
 def save_to_google_drive(image, prompt):
-    """Save image to Google Drive"""
+    """Save image to Google Drive Shared Drive with detailed debugging"""
     if not drive_service:
-        print("Google Drive not configured, skipping save")
+        print("❌ Google Drive service not available, skipping save")
+        return None
+        
+    if not SHARED_DRIVE_ID:
+        print("❌ Shared Drive ID not configured, skipping save")
         return None
         
     try:
+        print(f"🔄 Attempting to save image to Shared Drive: {SHARED_DRIVE_ID}")
+        
         # Create a filename with timestamp
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         safe_prompt = "".join(c for c in prompt[:30] if c.isalnum() or c in (' ', '-', '_')).rstrip()
         filename = f"storybook_{timestamp}_{safe_prompt}.png"
         
+        print(f"📁 Filename: {filename}")
+        
         # Convert image to bytes
         img_bytes = io.BytesIO()
         image.save(img_bytes, format='PNG')
@@ -116,22 +167,30 @@ def save_to_google_drive(image, prompt):
         file_metadata = {
             'name': filename,
             'mimeType': 'image/png',
-            'parents': ['root']  # Save to root folder, or specify folder ID
+            'parents': [SHARED_DRIVE_ID]  # Save to Shared Drive
         }
         
-        # Upload to Google Drive
+        print(f"📋 File metadata: {file_metadata}")
+        
+        # Upload to Google Drive with supportsAllDrives=True
         media = MediaIoBaseUpload(img_bytes, mimetype='image/png', resumable=True)
+        
+        print("⬆️  Starting upload to Google Drive...")
         file = drive_service.files().create(
             body=file_metadata,
             media_body=media,
+            supportsAllDrives=True,  # CRITICAL FOR SHARED DRIVES
             fields='id, webViewLink'
         ).execute()
         
-        print(f"Image saved to Google Drive: {file.get('webViewLink')}")
-        return file.get('webViewLink')
+        drive_link = file.get('webViewLink')
+        print(f"✅ Image saved to Google Drive: {drive_link}")
+        return drive_link
         
     except Exception as e:
-        print(f"Failed to save to Google Drive: {e}")
+        print(f"❌ Failed to save to Google Drive: {str(e)}")
+        import traceback
+        traceback.print_exc()
         return None
 
 # Request model