Update app.py

app.py

@@ -139,9 +139,35 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
         raise gr.Error("something went wrong")
     return "Model uploaded!"
 
-def check_civit_link(url):
+def check_civit_link(profile: Optional[gr.OAuthProfile], url):
     info, _ = process_url(url, download_files=False)
-    return info["creator"]
+    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")
+        
+        # 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)