ai-app-factory / src /getGradioApp.mts
jbilcke-hf's picture
jbilcke-hf HF staff
working on an improved version
296b17c
import { gradioDoc } from "./gradioDoc.mts"
export function getGradioApp(prompt: string) {
const prefix = "# In app.py:\n```"
const instructions = [
{
role: "system",
content: [
`You are a Python developer, expert at crafting Gradio applications to deploy to Hugging Face. You must generate valid Python code. Don't forget the requirements.txt files!`,
`Here is an example of a minimal Gradio application:`,
gradioDoc
].filter(item => item).join("\n")
},
{
role: "user",
content: `Please write, file by file, the source code for a Gradio project.
DO NOT WRITE AN EXAMPLE! WRITE THE FULL CODE, NOT AN EXAMPLE.
You must not leave any TODO in the code.
You MUST use the following Python modules:
- gradio (version 5.23.3)
- torch (version 2.6.0)
- accelerate (version 1.6.0)
You are free to use (if necessary) the following Python modules. In tha case, don't specify a version, just use them as-is so it uses the latest one. Make sure to add them to the requirements.txt:
- numpy
- matplotlib
- diffusers
- transformers
- huggingface_hub
Don't forget to write a README.md with the following header, or else you will be FIRED:
\`\`\`
---
license: apache-2.0
title: <app name>
sdk: gradio
sdk_version: 5.23.3
app_file: app.py
emoji: πŸ‘€
colorFrom: green
colorTo: blue
---
\`\`\`
The app is about: ${prompt}`,
}
]
return { prefix, files: [], instructions }
}