scriptoriumadmin/code2video
1
0
Fork 0
mirror of https://github.com/showlab/Code2Video.git synced 2026-10-08 12:47:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge cbcfe93156 into c3dd611b55

Browse source
This commit is contained in:
Sense_wang 2026-05-12 13:54:51 +08:00 • committed by GitHub
commit fb5fdd2389
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/external_assets.py
View file

@ -118,11 +118,11 @@ class SmartSVGDownloader:
response = self.api_function(prompt, max_tokens=100)[0] response = self.api_function(prompt, max_tokens=100)[0]
try: try:
content = response.candidates[0].content.parts[0].text content = response.candidates[0].content.parts[0].text
except: except Exception:
content = response.choices[0].message.content content = response.choices[0].message.content
elements = [line.strip().lower() for line in content.strip().split("\n") if line.strip()] elements = [line.strip().lower() for line in content.strip().split("\n") if line.strip()]
return list(dict.fromkeys(elements))[:4] return list(dict.fromkeys(elements))[:4]
except: except Exception:
return [] return []
def _check_cache(self, element: str) -> Optional[str]: def _check_cache(self, element: str) -> Optional[str]:
@ -162,7 +162,7 @@ class SmartSVGDownloader:
filepath = self.assets_dir / f"{element}.png" filepath = self.assets_dir / f"{element}.png"
filepath.write_bytes(img_resp.content) filepath.write_bytes(img_resp.content)
return str(filepath.absolute()) return str(filepath.absolute())
except: except Exception:
return None return None
def _download_iconify(self, element: str) -> Optional[str]: def _download_iconify(self, element: str) -> Optional[str]:
@ -178,7 +178,7 @@ class SmartSVGDownloader:
filepath = self.assets_dir / f"{element}.svg" filepath = self.assets_dir / f"{element}.svg"
filepath.write_text(svg_resp.text, encoding="utf-8") filepath.write_text(svg_resp.text, encoding="utf-8")
return str(filepath.absolute()) return str(filepath.absolute())
except: except Exception:
return None return None
def _enhance_animations(self, animations: List[str], assets: Dict[str, str]) -> List[str]: def _enhance_animations(self, animations: List[str], assets: Dict[str, str]) -> List[str]: