Update app.py

app.py

@@ -90,6 +90,11 @@ def setup_google_drive():
 # 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"""
     if not drive_service: