| """Application-wide constants, regex patterns, language options, and system prompt.""" |
|
|
| from __future__ import annotations |
|
|
| import re |
|
|
| |
|
|
| APP_TITLE = "Fullstack Code Builder" |
| MODEL_URL = "https://huggingface.co/openbmb/MiniCPM5-1B" |
|
|
| |
|
|
| MODEL_CONFIGS = { |
| "minicpm5-1b": { |
| "id": "openbmb/MiniCPM5-1B", |
| "name": "MiniCPM5-1B", |
| "type": "text", |
| "description": "Text-only, fast code generation", |
| "auto_class": "AutoModelForCausalLM", |
| "tokenizer_class": "AutoTokenizer", |
| "size_gb": 2.17, |
| }, |
| "minicpm-v-4.6": { |
| "id": "openbmb/MiniCPM-V-4.6", |
| "name": "MiniCPM-V-4.6", |
| "type": "vlm", |
| "description": "Vision + Text, image understanding & code", |
| "auto_class": "AutoModelForImageTextToText", |
| "processor_class": "AutoProcessor", |
| "size_gb": 2.8, |
| }, |
| } |
|
|
| DEFAULT_MODEL_KEY = "minicpm5-1b" |
|
|
| |
| MODEL_ID = MODEL_CONFIGS[DEFAULT_MODEL_KEY]["id"] |
|
|
| |
|
|
| DEFAULT_TEMPERATURE = 0.4 |
| DEFAULT_MAX_TOKENS = 2048 |
| PY_TIMEOUT_S = 15 |
| GRADIO_TIMEOUT_S = 30 |
| PY_MEM_LIMIT_MB = 1024 |
| MAX_STDIO_CHARS = 16_000 |
| OUTPUT_PNG = "output.png" |
|
|
| |
|
|
| THINKING_BLOCK_RE = re.compile( |
| r"<\s*think\s*>.*?<\s*/\s*think\s*>", re.IGNORECASE | re.DOTALL |
| ) |
| CODE_BLOCK_RE = re.compile( |
| r"```([a-zA-Z0-9_+.#-]*)\s*\n(.*?)```", re.DOTALL |
| ) |
| FILE_BLOCK_RE = re.compile( |
| r"@@FILE:\s*(.+?)@@\s*\n(.*?)(?=@@FILE:|@@END@@)", re.DOTALL |
| ) |
|
|
| |
|
|
| LANGUAGE_OPTIONS: list[tuple[str, list[str]]] = [ |
| ("Python", ["Gradio", "Flask", "Django", "FastAPI", "Streamlit", "Plain Python"]), |
| ("JavaScript", ["React", "Vue.js", "Next.js", "Express.js", "Node.js", "Vanilla JS"]), |
| ("TypeScript", ["React", "Next.js", "Express.js", "NestJS"]), |
| ("HTML/CSS/JS", ["Tailwind CSS", "Bootstrap", "Vanilla"]), |
| ("Java", ["Spring Boot", "Maven", "Gradle"]), |
| ("Go", ["Gin", "Fiber", "Echo", "Plain Go"]), |
| ("Rust", ["Actix", "Axum", "Rocket"]), |
| ("PHP", ["Laravel", "Symfony", "Plain PHP"]), |
| ("Ruby", ["Rails", "Sinatra"]), |
| ("C#", ["ASP.NET", "Blazor"]), |
| ("Swift", ["Vapor", "SwiftUI"]), |
| ("Kotlin", ["Ktor", "Spring Boot"]), |
| ] |
|
|
| LANGUAGE_MAP: dict[str, list[str]] = {lang: frameworks for lang, frameworks in LANGUAGE_OPTIONS} |
|
|
| |
|
|
| SYSTEM_PROMPT = """You are a code generator. Output ONLY the code. No thinking, no explanation, no commentary. |
| |
| CRITICAL RULES: |
| - Do NOT use <think> or <thinking> tags. Do NOT reason aloud. Just output code directly. |
| - Do NOT write explanations before or after code. Just output the code. |
| - If you must explain something, keep it to ONE short sentence. |
| |
| When the user asks you to build an application: |
| 1. Generate complete, working code - not snippets or pseudocode |
| 2. Include all necessary files for the project to run |
| 3. Add proper error handling and comments |
| 4. For web apps, make the UI responsive and modern |
| 5. For Gradio apps, use gradio library and create a complete working app with gr.Interface or gr.Blocks |
| |
| FILE OUTPUT FORMAT - IMPORTANT: |
| When generating multi-file projects, wrap each file in this format: |
| @@FILE: path/to/file.ext@@ |
| (file content here) |
| @@FILE: path/to/another/file.ext@@ |
| (another file content here) |
| @@END@@ |
| |
| For single-file code, use standard markdown fenced blocks: |
| ```python for Python |
| ```html for HTML/CSS/JS |
| ```javascript for JavaScript |
| ```typescript for TypeScript |
| etc. |
| |
| JAVASCRIPT / TYPESCRIPT PROJECTS: |
| For React, Next.js, Vue.js, Express, NestJS, or any JS/TS framework: |
| - ALWAYS use the @@FILE: multi-file format |
| - Include a package.json with name, version, scripts, and dependencies |
| - Include all source files (src/App.jsx, src/index.js, etc.) |
| - For React+Vite: include vite.config.js and index.html |
| - For Next.js: include next.config.js with output: 'standalone' |
| - For Express: main entry is index.js with app.listen(7860) |
| - Server ports MUST be 7860 and bind to 0.0.0.0 |
| - Do NOT include node_modules or lock files |
| |
| When generating web apps with HTML/CSS/JS, return a single self-contained HTML document with all CSS and JavaScript inline. Make the page fully responsive: html/body at margin:0 and 100% width/height, use flexbox/grid layouts, and size any canvas to its container. |
| |
| When generating Gradio apps, create a complete app.py with: |
| - import gradio as gr |
| - Define the interface using gr.Interface() or gr.Blocks() |
| - Call iface.launch(server_name="0.0.0.0", server_port=7860) at the end |
| - Include all necessary processing logic inline |
| |
| For Python, prefer standard library or common packages. Do not use network calls, subprocesses, shell commands, or long-running loops in demo code (except Gradio apps which are server-based). |
| |
| If web search results are provided in the context, use them to inform your code generation. Incorporate relevant information from the search results into the generated code. |
| |
| If the user provides an image, analyze it and generate code based on what you see in the image. For example: replicate a UI from a screenshot, generate code from a wireframe, or build an app described in a document. |
| """ |
|
|
| |
|
|
| EXAMPLE_PROMPTS: list[tuple[str, str, str, str]] = [ |
| ( |
| "🎨 Gradio Image Filter", |
| "Create a Gradio app that lets users upload an image and apply filters like grayscale, blur, sepia, and edge detection using PIL. Show the original and filtered images side by side.", |
| "Python", |
| "Gradio", |
| ), |
| ( |
| "🤖 Gradio Chat App", |
| "Build a Gradio chatbot app with gr.Blocks that has a chat interface, a text input, and a send button. Include a simple echo bot that repeats the user's message with a fun twist.", |
| "Python", |
| "Gradio", |
| ), |
| ( |
| "🌐 React Todo App", |
| "Build a React todo application with add, delete, mark complete, and filter functionality. Use modern hooks and a clean responsive UI.", |
| "JavaScript", |
| "React", |
| ), |
| ( |
| "🐍 Flask API", |
| "Create a Flask REST API for a book library with CRUD operations, in-memory storage, and proper error handling.", |
| "Python", |
| "Flask", |
| ), |
| ( |
| "🎨 Landing Page", |
| "Build a modern landing page for a SaaS product with a hero section, features grid, pricing cards, and a footer. Use Tailwind-style CSS.", |
| "HTML/CSS/JS", |
| "Vanilla", |
| ), |
| ( |
| "📊 Dashboard", |
| "Create an interactive data dashboard with charts (bar, line, pie), a sidebar navigation, and summary cards. All in a single HTML file.", |
| "HTML/CSS/JS", |
| "Vanilla", |
| ), |
| ] |
|
|
