Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

print system prompt info in the backend logs

Browse Source
This commit is contained in:
Abi Raja 2023-11-28 14:33:03 -05:00
parent dfe8ae4164
commit 8c08d992fd
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/prompts.py
View File

@ -84,12 +84,17 @@ Generate code for a web page that looks exactly like this.
def assemble_prompt(image_data_url, output_settings: dict, result_image_data_url=None): def assemble_prompt(image_data_url, output_settings: dict, result_image_data_url=None):
# Set the system prompt based on the output settings # Set the system prompt based on the output settings
chosen_prompt_name = "tailwind"
system_content = TAILWIND_SYSTEM_PROMPT system_content = TAILWIND_SYSTEM_PROMPT
if output_settings["css"] == "bootstrap": if output_settings["css"] == "bootstrap":
chosen_prompt_name = "bootstrap"
system_content = BOOTSTRAP_SYSTEM_PROMPT system_content = BOOTSTRAP_SYSTEM_PROMPT
if output_settings["js"] == "react": if output_settings["js"] == "react":
chosen_prompt_name = "react-tailwind"
system_content = REACT_TAILWIND_SYSTEM_PROMPT system_content = REACT_TAILWIND_SYSTEM_PROMPT
print("Using system prompt:", chosen_prompt_name)
user_content = [ user_content = [
{ {
"type": "image_url", "type": "image_url",