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.

You MUST use the following Python modules:
- gradio (version 3.39.0)

You are free to use (if necessary) the following Python modules:
- numpy
- matplotlib
- pytorch
- diffusers
- transformers

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: 3.39.0
app_file: app.py
emoji: 👀
colorFrom: green
colorTo: blue
---
\`\`\`

The app is about: ${prompt}`,
    }
  ]

  return { prefix, files: [], instructions }
}