CalvinBackup/ScreenCoder_UI2Code
1
0
Fork 0
mirror of https://github.com/leigest519/ScreenCoder.git synced 2026-02-12 17:52:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update html_generator.py

Browse Source 
Add qwen, gpt, gemini functions
This commit is contained in:
JimmyZhengyz
2025-07-30 13:08:54 +08:00
committed by GitHub
parent ef42f6b32e
commit 8cc9af06b9
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
html_generator.py
View File

@@ -1,4 +1,4 @@
from utils import encode_image, Doubao, Qwen_2_5_VL
from utils import encode_image, Doubao, Qwen, GPT, Gemini
from PIL import Image
import bs4
from threading import Thread
@@ -376,9 +376,12 @@ if __name__ == "__main__":
generate_html(root, 'data/tmp/test2_layout.html')
# Initialize the bot
# Change your model & API ket path according to your needs
bot = Doubao("doubao_api.txt", model = "doubao-1.5-thinking-vision-pro-250428")
# bot = Qwen_2_5_VL("qwen_api.txt", model="qwen2.5-vl-72b-instruct")
# bot = Qwen("qwen_api.txt", model="qwen2.5-vl-72b-instruct")
# bot = GPT("gpt_api.txt", model="gpt-4o")
# bot = Gemini("gemini_api.txt", model="gemini-1.5-flash-latest")
# Generate code for each component
# code_dict = generate_code(root, img_path, bot)
@@ -388,4 +391,4 @@ if __name__ == "__main__":
code_substitution('data/tmp/test2_layout.html', code_dict)
# Refine the html file
# html_refinement('data/tmp/test1_layout.html', 'data/tmp/test1_layout_refined.html', img_path, bot)
# html_refinement('data/tmp/test1_layout.html', 'data/tmp/test1_layout_refined.html', img_path, bot)