Update app.py

app.py

@@ -15,7 +15,8 @@ TRUSTED_UPLOADERS = ["KappaNeuro", "CiroN2022", "multimodalart", "Norod78", "joa
 def get_json_data(url):
     url_split = url.split('/')
     print("url_split: ", url_split)
-    api_url = f"https://civitai.com/api/v1/models/{url_split[4]}?token={os.environ['CIVITAI_API']}"
+    api_url = f"https://civitai.com/api/v1/models/{url_split[4]}?
+    token={os.environ['CIVITAI_API']}"
     try:
         response = requests.get(api_url)
         response.raise_for_status()
@@ -123,14 +124,26 @@ def download_files(info, folder="."):
     return downloaded_files
 
 def download_file(url, filename, folder="."):
+    headers = {}
     try:
-        response = requests.get(url)
+        response = requests.get(url, headers=headers)
         response.raise_for_status()
-        with open(f"{folder}/{filename}", 'wb') as f:
-            f.write(response.content)
+    except requests.exceptions.HTTPError as e:
+        if api_key and response.status_code == 401: 
+            headers['Authorization'] = f'Bearer {os.environ['CIVITAI_API']}'
+            try:
+                response = requests.get(url, headers=headers)
+                response.raise_for_status()
+            except requests.exceptions.RequestException as e:
+                raise gr.Error(f"Error downloading file: {e}")
+        else:
+            raise gr.Error(f"Error downloading file: {e}")
     except requests.exceptions.RequestException as e:
         raise gr.Error(f"Error downloading file: {e}")
 
+    with open(f"{folder}/{filename}", 'wb') as f:
+        f.write(response.content)
+
 def process_url(url, profile, do_download=True, folder="."):
     json_data = get_json_data(url)
     if json_data: