Update app.py

app.py

@@ -65,9 +65,8 @@ def download_file(url, filename, folder="."):
         response.raise_for_status()
         with open(f"{folder}/{filename}", 'wb') as f:
             f.write(response.content)
-        print(f"{filename} downloaded.")
     except requests.exceptions.RequestException as e:
-        print(f"Error downloading file: {e}")
+        raise gr.Error(f"Error downloading file: {e}")
 
 def process_url(url, do_download=True, folder="."):
     json_data = get_json_data(url)
@@ -124,12 +123,8 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
     
     folder = str(uuid.uuid4())
     os.makedirs(folder, exist_ok=False)
-    info, downloaded_files = process_url(url, folder)
-    create_readme(info, downloaded_files, folder)
-    print(os.path.abspath("."))
-    print(os.listdir("."))
-    print(os.path.abspath(folder))
-    print(os.listdir(folder))
+    info, downloaded_files = process_url(url, folder=folder)
+    create_readme(info, downloaded_files, folder=folder)
     try:
         api = HfApi(token=os.environ["HUGGING_FACE_HUB_TOKEN"])
         username = api.whoami()["name"]
@@ -144,7 +139,6 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
         transfer_repos = gr.load("multimodalart/transfer_repos", hf_token=os.environ["HUGGING_FACE_HUB_TOKEN"], src="spaces")
         user_repo_id = f"{profile.preferred_username}/{slug_name}"
         response_code = transfer_repos(repo_id, user_repo_id)
-        print("response_code", response_code)
         i = 0
         while response_code != "200":
             if response_code == "404":
@@ -190,12 +184,10 @@ def get_creator(username):
 def extract_huggingface_username(username):
     data = get_creator(username)
     links = data.get('result', {}).get('data', {}).get('json', {}).get('links', [])
-    print(links)
     for link in links:
         url = link.get('url', '')
         if url.startswith('https://huggingface.co/') or url.startswith('https://www.huggingface.co/'):
             username = url.split('/')[-1]
-            print("Extracted username:", username)
             return username
 
     return None
@@ -204,11 +196,7 @@ def extract_huggingface_username(username):
 def check_civit_link(profile: Optional[gr.OAuthProfile], url):
     info, _ = process_url(url, do_download=False)
     hf_username = extract_huggingface_username(info['creator'])
-    print(hf_username)
     attributes_methods = dir(profile)
-    for attribute in attributes_methods:
-        if not attribute.startswith('__'):
-            print(f"{attribute}: {getattr(profile, attribute)}")
     if(not hf_username):
         no_username_text = 'Oops, your CivitAI profile seems to not have information about your Hugging Face account. Please visit <a href="https://civitai.com/user/account">https://civitai.com/user/account</a> and include it there<br><img width="60%" src="https://i.imgur.com/hCbo9uL.png" />'
         return no_username_text, gr.update(), gr.update(visible=True)