Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -540,48 +540,6 @@ def save(Huggingface_Write_token):
         else:
             print('Nothing to backup')
         
-
-
-
-def getsrc(url):
-
-    parsed_url = urlparse(url)
-    
-    if parsed_url.netloc == 'civitai.com':
-        src='civitai'
-    elif parsed_url.netloc == 'drive.google.com':
-        src='gdrive'
-    elif parsed_url.netloc == 'huggingface.co':
-        src='huggingface'
-    else:
-        src='others'
-    return src
-
-
-
-def get_name(url, gdrive):
-
-    from gdown.download import get_url_from_gdrive_confirmation
-
-    if not gdrive:
-        response = requests.get(url, allow_redirects=False)
-        if "Location" in response.headers:
-            redirected_url = response.headers["Location"]
-            quer = parse_qs(urlparse(redirected_url).query)
-            if "response-content-disposition" in quer:
-                disp_val = quer["response-content-disposition"][0].split(";")
-                for vals in disp_val:
-                    if vals.strip().startswith("filename="):
-                        filenm=unquote(vals.split("=", 1)[1].strip())
-                        return filenm.replace("\"","")
-    else:
-        headers = {"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36"}
-        lnk="https://drive.google.com/uc?id={id}&export=download".format(id=url[url.find("/d/")+3:url.find("/view")])
-        res = requests.session().get(lnk, headers=headers, stream=True, verify=True)
-        res = requests.session().get(get_url_from_gdrive_confirmation(res.text), headers=headers, stream=True, verify=True)
-        content_disposition = six.moves.urllib_parse.unquote(res.headers["Content-Disposition"])
-        filenm = re.search('attachment; filename="(.*?)"', content_disposition).groups()[0]
-        return filenm   
 def download_and_install_config():
     target_directory = "/workspace/sd/stable-diffusion-webui/"
     config_url = (
@@ -739,7 +697,72 @@ def InstallWildCards():
 
     # Возвращаемся в исходную рабочую директорию
     os.chdir("/workspace")
+def getsrc(url):
 
+    parsed_url = urlparse(url)
+
+    if parsed_url.netloc == "civitai.com":
+        src = "civitai"
+    elif parsed_url.netloc == "drive.google.com":
+        src = "gdrive"
+    elif parsed_url.netloc == "huggingface.co":
+        src = "huggingface"
+    else:
+        src = "others"
+    return src
+
+
+def get_true_name(url):
+    response = requests.get(url)
+    soup = BeautifulSoup(response.text, "html.parser")
+    title_tag = soup.find("title")
+    if title_tag:
+        title_text = title_tag.text
+        # Извлечение имени файла из тега title (предполагая, что имя файла указано в теге title)
+        file_name = title_text.split(" - ")[0]
+        return file_name
+    else:
+        raise RuntimeError("Could not find the title tag in the HTML")
+
+
+def get_name(url, gdrive):
+
+    from gdown.download import get_url_from_gdrive_confirmation
+
+    if not gdrive:
+        response = requests.get(url, allow_redirects=False)
+        if "Location" in response.headers:
+            redirected_url = response.headers["Location"]
+            quer = parse_qs(urlparse(redirected_url).query)
+            if "response-content-disposition" in quer:
+                disp_val = quer["response-content-disposition"][0].split(";")
+                for vals in disp_val:
+                    if vals.strip().startswith("filename="):
+                        filenm = unquote(vals.split("=", 1)[1].strip())
+                        return filenm.replace('"', "")
+    else:
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36"
+        }
+        lnk = "https://drive.google.com/uc?id={id}&export=download".format(
+            id=url[url.find("/d/") + 3 : url.find("/view")]
+        )
+        res = requests.session().get(lnk, headers=headers, stream=True, verify=True)
+        res = requests.session().get(
+            get_url_from_gdrive_confirmation(res.text),
+            headers=headers,
+            stream=True,
+            verify=True,
+        )
+        content_disposition = six.moves.urllib_parse.unquote(
+            res.headers["Content-Disposition"]
+        )
+        filenm = (
+            re.search(r"filename\*=UTF-8''(.*)", content_disposition)
+            .groups()[0]
+            .replace(os.path.sep, "_")
+        )
+        return filenm
 
 
 def done():