Update app.py

app.py

@@ -6,14 +6,7 @@ from slugify import slugify
 import gradio as gr
 import uuid
 from typing import Optional
-
-from selenium import webdriver
-from webdriver_manager.chrome import ChromeDriverManager
-from selenium.webdriver.common.by import By
-from selenium.webdriver.support.ui import WebDriverWait
-from selenium.webdriver.support import expected_conditions as EC
-
-driver = webdriver.Chrome(ChromeDriverManager().install())
+import json
 
 
 def get_json_data(url):
@@ -148,35 +141,54 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
         raise gr.Error("something went wrong")
     return "Model uploaded!"
 
+
+def get_creator(username):
+    url = f"https://civitai.com/api/trpc/user.getCreator?input=%7B%22json%22%3A%7B%22username%22%3A%22{username}%22%2C%22authed%22%3Atrue%7D%7D"
+    headers = {
+        "authority": "civitai.com",
+        "accept": "*/*",
+        "accept-language": "en-BR,en;q=0.9,pt-BR;q=0.8,pt;q=0.7,es-ES;q=0.6,es;q=0.5,de-LI;q=0.4,de;q=0.3,en-GB;q=0.2,en-US;q=0.1,sk;q=0.1",
+        "content-type": "application/json",
+        "cookie": f"{os.environ["COOKIE_INFO"]}",
+        "if-modified-since": "Tue, 22 Aug 2023 07:18:52 GMT",
+        "referer": f"https://civitai.com/user/{username}/models",
+        "sec-ch-ua": "\"Not.A/Brand\";v=\"8\", \"Chromium\";v=\"114\", \"Google Chrome\";v=\"114\"",
+        "sec-ch-ua-mobile": "?0",
+        "sec-ch-ua-platform": "macOS",
+        "sec-fetch-dest": "empty",
+        "sec-fetch-mode": "cors",
+        "sec-fetch-site": "same-origin",
+        "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36"
+    }
+    response = requests.get(url, headers=headers)
+
+    return json.loads(response.text)
+
+def extract_huggingface_username(username):
+    data = get_creator(username)
+    links = data.get('result', {}).get('data', {}).get('json', {}).get('links', [])
+
+    for link in links:
+        url = link.get('url', '')
+        if url.startswith('https://www.huggingface.co/'):
+            return url.split('/')[-1]
+
+    return None
+
+
+
 def check_civit_link(profile: Optional[gr.OAuthProfile], url):
     info, _ = process_url(url, download_files=False)
-    url_creator = f"https://civitai.com/user/{info['creator']}/models" 
-    # Open the target URL
-    driver.get(url_creator)
-    
-    # Define the XPath expression
-    xpath_expression = "//a[contains(@class, 'mantine-UnstyledButton-root') and contains(@class, 'mantine-ActionIcon-root') and contains(@class, 'mantine-ubxmi3') and starts-with(@href, 'https://huggingface.co/')]"
-    
-    # Find the element using the XPath expression
-    try:
-        element = WebDriverWait(driver, 10).until(
-            EC.presence_of_element_located((By.XPATH, xpath_expression))
-        )
-        
-        # Extract the href attribute
-        href = element.get_attribute("href")
+    hf_username = extract_huggingface_username(info['creator'])
+    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 src="https://i.imgur.com/hCbo9uL.png" />'
+        return no_username_text, gr.update()
+    if(profile.name != hf_username):
+        unmatched_username_text = 'Oops, the Hugging Face account in your CivitAI profile seems to be different than the one your are using here. Please visit <a href="https://civitai.com/user/account">https://civitai.com/user/account</a> and update it there<br><img src="https://i.imgur.com/hCbo9uL.png" />'
+        return unmatched_username_text, gr.update()
+    else:
+        return '', gr.update(interactive=True)
         
-        # Extract the part after 'https://huggingface.co/'
-        extracted_part = href.replace("https://huggingface.co/", "")
-    
-    except Exception as e:
-        print("Element not found or error occurred:", e)
-    
-    finally:
-        driver.quit()
-    
-    return extracted_part == profile.name
-    
 def swap_fill(profile: Optional[gr.OAuthProfile]):
     if profile is None:
         return gr.update(visible=True), gr.update(visible=False)
@@ -226,11 +238,11 @@ with gr.Blocks(css=css) as demo:
                         )
                         #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")
+        submit_button_civit = gr.Button("Upload model to Hugging Face", interactive=False)
         output = gr.Textbox(label="Output progress")
    
     demo.load(fn=swap_fill, outputs=[disabled_area, enabled_area])
-    submit_source_civit.change(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions])
+    submit_source_civit.change(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions, submit_button_civit])
     submit_button_civit.click(fn=upload_civit_to_hf, inputs=[submit_source_civit], outputs=[output])
     
 demo.queue()