Update app.py

app.py

@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
 import os
 import openai
 from sklearn.preprocessing import LabelEncoder
-import requests 
+import requests  # Add this at the top with other imports
 from io import BytesIO
 import gdown
 
@@ -175,12 +175,11 @@ 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"
-            url = f"https://drive.google.com/uc?id={file_id}&confirm=t"
-            gdown.download(url, output, quiet=False, fuzzy=True)
+            gdown.download(url, output, quiet=False)
             
             with open(output, 'rb') as f:
                 content = f.read()
@@ -191,15 +190,7 @@ def download_file_from_google_drive(file_id):
             
     except Exception as e:
         st.error(f"Error downloading from Google Drive: {str(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)}")
+        raise e
 
 @st.cache_data
 def load_datasets():