update some exception

Author: WithHades

background.png

@@ -47,7 +47,7 @@ def run(self):
         self.signal.emit(text)
         '''
 
-        text = getPredictText(self.text.replace('\n', ''), length=50)
+        text = getPredictText(self.text.replace('\n', ''), length=500)
         self.signal.emit(text)
 
 

edit.py

@@ -24,3 +24,7 @@ def getPredictText(text: str, length: int = 200) -> str:
     # 使用tokenizer将生成的id转为文本
     generatedText = tokenizer.convert_ids_to_string(ids)
     return generatedText
+
+
+getPredictText('开始预测模型会先初始化一下, 抵消掉这个时间')
+

generation.py

@@ -45,7 +45,12 @@ def run(self):
         screensize = (800, 600)
         videoClips = []
         for i, section in enumerate(self.sections):
+
             imgPath, text = section[0], section[1]
+            if len(text) >= 1:
+                mark = text[-1]
+                if mark == '$':
+                    text = text[:-1]
             text = text.split(r'\\')
             text = '\\'.join([x.replace(r'\n', '\n') for x in text])
             text = text.strip().split('\n')
@@ -64,9 +69,11 @@ def run(self):
                 clip = ImageClip(imgPath)
 
             # 设置一下图片/gif大小
-            width, height = clip.size
-            width, height = resizeImg(width, height)
-            clip = clip.resize((width, height))
+            if mark != '$':
+                width, height = clip.size
+                width, height = resizeImg(width, height)
+                clip = clip.resize((width, height))
+
 
             # 考虑到每张表情包可能对应多句字幕
             for txt in text:

main.py

@@ -134,7 +134,10 @@ def getBaiduImgPath(text: str) -> [Iterable, None]:
     res = requests.get(url, headers=headers, verify=False)
     if res.status_code != 200:
         return None
-    jsonData = json.loads(res.text)
+    try:
+        jsonData = json.loads(res.text)
+    except:
+        return None
     if 'data' not in jsonData:
         return None
     data = jsonData['data']
@@ -182,7 +185,10 @@ def getDoutulaImgPath(text: str) -> [Iterable, None]:
     randomPics = soup.find_all('a', attrs={'class': 'col-xs-6 col-md-2'})
     for pic in randomPics:
         imgUrl = pic.find('img', attrs={'referrerpolicy': 'no-referrer'})['data-original']
-        res = requests.get(imgUrl)
+        try:
+            res = requests.get(imgUrl)
+        except:
+            return None
         if res.status_code != 200:
             continue
         path = os.path.join('tmp', imgUrl[imgUrl.rfind('/') + 1:])