Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -10,6 +10,7 @@ from torch.hub import download_url_to_file
 from urllib.parse import urlparse, parse_qs, unquote
 import re
 import six
+import gdown
 
 from urllib.request import urlopen, Request
 import tempfile
@@ -217,8 +218,7 @@ def mdl(Original_Model_Version, Path_to_MODEL, MODEL_LINK):
 
     return model
 def modeldwn(model_LINK):
-    import gdown
-    
+
     if model_LINK=='':
         print('Nothing to do')
     else:
@@ -235,7 +235,7 @@ def modeldwn(model_LINK):
             else:
               print('Model already exists')
         elif src=='gdrive':
-            modelname=get_name(model_LINK, True)
+            modelname=get_true_name(model_LINK)
             loramodel=f'/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}'
             if not os.path.exists(loramodel):
               gdown.download(url=model_LINK, output=loramodel, quiet=False, fuzzy=True)
@@ -257,8 +257,7 @@ def modeldwn(model_LINK):
           print('Wrong link, check that the link is valid')
 
 def loradwn(LoRA_LINK):
-    import gdown
-    
+
     if LoRA_LINK=='':
         print('Nothing to do')
     else:
@@ -275,7 +274,7 @@ def loradwn(LoRA_LINK):
             else:
               print('Model already exists')
         elif src=='gdrive':
-            modelname=get_name(LoRA_LINK, True)
+            modelname=get_true_name(LoRA_LINK)
             loramodel=f'/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}'
             if not os.path.exists(loramodel):
               gdown.download(url=LoRA_LINK, output=loramodel, quiet=False, fuzzy=True)
@@ -498,7 +497,31 @@ def getsrc(url):
         src='others'
     return src
 
+def get_true_name(url):
+    import requests
+    from bs4 import BeautifulSoup
+
+    # Отправляем запрос на страницу предпросмотра файла
+    response = requests.get(url)
+
+    # Проверяем, успешно ли выполнен запрос
+    if response.status_code != 200:
+        print(f"Failed to retrieve the page: {response.status_code}")
+        return None
 
+    # Используем BeautifulSoup для парсинга HTML
+    soup = BeautifulSoup(response.text, 'html.parser')
+
+    # Ищем тег <title>, который может содержать имя файла
+    title_tag = soup.find('title')
+    if title_tag:
+        # Имя файла может быть в формате "file_name.extension - Google Drive" в теге <title>
+        # Поэтому мы разделяем строку по ' - ' и берем первую часть
+        file_name = title_tag.text.split(' - ')[0]
+        return file_name
+    else:
+        print("Failed to find the title tag")
+        return None
 
 def get_name(url, gdrive):