Update app.py

app.py

@@ -175,11 +175,12 @@ def create_brand_categories():
 @st.cache_resource
 def download_file_from_google_drive(file_id):
     """Downloads a file from Google Drive using gdown."""
-    url = f"https://drive.google.com/uc?id={file_id}"
     try:
         with st.spinner('Downloading from Google Drive...'):
+            # Use gdown with confirmation flag
             output = f"temp_{file_id}.pkl"
-            gdown.download(url, output, quiet=False)
+            url = f"https://drive.google.com/uc?id={file_id}&confirm=t"
+            gdown.download(url, output, quiet=False, fuzzy=True)
             
             with open(output, 'rb') as f:
                 content = f.read()
@@ -190,7 +191,15 @@ def download_file_from_google_drive(file_id):
             
     except Exception as e:
         st.error(f"Error downloading from Google Drive: {str(e)}")
-        raise e
+        try:
+            url = f"https://drive.google.com/uc?export=download&id={file_id}&confirm=t"
+            response = requests.get(url)
+            if response.status_code == 200:
+                return response.content
+            else:
+                raise Exception(f"Failed to download with status code: {response.status_code}")
+        except Exception as e2:
+            raise Exception(f"Both download methods failed. Error: {str(e2)}")
 
 @st.cache_data
 def load_datasets():