Update app.py

app.py

@@ -67,22 +67,25 @@ def download_file(url, filename, folder="."):
     except requests.exceptions.RequestException as e:
         print(f"Error downloading file: {e}")
 
-def process_url(url, folder="."):
+def process_url(url, download_files=True, folder="."):
     json_data = get_json_data(url)
     if json_data:
         if check_nsfw(json_data):
             info = extract_info(json_data)
             if info:
-                downloaded_files = download_files(info, folder)
+                if(download_files):
+                    downloaded_files = download_files(info, folder)
+                else
+                    downloaded_files = []
                 return info, downloaded_files
             else:
-                print("No model met the criteria.")
+                raise gr.Error("Only SDXL LoRAs are supported for now")
         else:
-            print("NSFW content found.")
+            raise gr.Error("This model has content tagged as unsafe by CivitAI")
     else:
-        print("Failed to get JSON data.")
+        raise gr.Error("Something went wrong in fetching CivitAI API")
 
-def create_readme(info, downloaded_files, is_author, folder="."):
+def create_readme(info, downloaded_files, is_author=True, folder="."):
     readme_content = ""
     original_url = f"https://civitai.com/models/{info['id']}"
     non_author_disclaimer = f'This model was originally uploaded on [CivitAI]({original_url}), by [{info["creator"]}](https://civitai.com/user/{info["creator"]}/models). The information below was provided by the author on CivitAI:'
@@ -113,14 +116,14 @@ widget:
         file.write(readme_content)
 
 
-def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, is_author, progress=gr.Progress(track_tqdm=True)):
+def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Progress(track_tqdm=True)):
     if not profile.name:
         return gr.Error("Are you sure you are logged in?")
     
     folder = str(uuid.uuid4())
     os.makedirs(folder, exist_ok=False)
     info, downloaded_files = process_url(url, folder)
-    create_readme(info, downloaded_files, False, folder)
+    create_readme(info, downloaded_files, folder)
     try:
         api = HfApi(token=hf_token)
         username = api.whoami()["name"]
@@ -136,6 +139,9 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, is_author, progr
         raise gr.Error("something went wrong")
     return "Model uploaded!"
 
+def check_civit_link(url):
+    info, _ = process_url(url, download_files=False)
+    return info["creator"]
 def swap_fill(profile: Optional[gr.OAuthProfile]):
     if profile is None:
         return gr.update(visible=True), gr.update(visible=False)
@@ -174,7 +180,7 @@ with gr.Blocks(css=css) as demo:
                             label="CivitAI model URL",
                             info="URL of the CivitAI model, make sure it is a SDXL LoRA",
                         )
-                        is_author = gr.Checkbox(label="Are you the model author?", info="If you are not the author, a disclaimer with information about the author and the CivitAI source will be added", value=False)
+                        #is_author = gr.Checkbox(label="Are you the model author?", info="If you are not the author, a disclaimer with information about the author and the CivitAI source will be added", value=False)
         submit_button_civit = gr.Button("Upload model to Hugging Face and submit")
         output = gr.Textbox(label="Output progress")
     with gr.Column(visible=False) as enabled_area:
@@ -183,11 +189,14 @@ with gr.Blocks(css=css) as demo:
                             label="CivitAI model URL",
                             info="URL of the CivitAI model, make sure it is a SDXL LoRA",
                         )
-                        is_author = gr.Checkbox(label="Are you the model author?", info="If you are not the author, a disclaimer with information about the author and the CivitAI source will be added", value=False)
+                        #is_author = gr.Checkbox(label="Are you the model author?", info="If you are not the author, a disclaimer with information about the author and the CivitAI source will be added", value=False)
+        instructions = gr.HTML("")
         submit_button_civit = gr.Button("Upload model to Hugging Face")
         output = gr.Textbox(label="Output progress")
+   
     demo.load(fn=swap_fill, outputs=[disabled_area, enabled_area])
-    submit_button_civit.click(fn=upload_civit_to_hf, inputs=[submit_source_civit, is_author], outputs=[output])
+    submit_source_civit.change(fn=check_civit_link, inputs=[submit_source_civit], output=[instructions])
+    submit_button_civit.click(fn=upload_civit_to_hf, inputs=[submit_source_civit], outputs=[output])
     
 demo.queue()
 demo.launch()