Spaces:
Sleeping
Sleeping
| """ | |
| Stage 1 — Intent Parser | |
| Model: DeepSeek R1 (reasoning model, ideal for structured analysis) | |
| Input: raw user prompt | |
| Output: structured intent dict used by the code generator | |
| """ | |
| import json | |
| import re | |
| from openai import OpenAI | |
| from config import CRUSOE_API_KEY, CRUSOE_BASE_URL, INTENT_MODEL | |
| client = OpenAI(api_key=CRUSOE_API_KEY, base_url=CRUSOE_BASE_URL) | |
| SYSTEM_PROMPT = """You are an expert at analysing AI demo requests and mapping them to the right app template. | |
| Given a user's demo idea, output a single JSON object with these fields: | |
| REQUIRED (all templates): | |
| - template_type: one of "chatbot" | "comparison" | "dashboard" | "form_wizard" | |
| - title: short, catchy demo title (max 8 words) | |
| - description: one-line description shown below the title | |
| - system_prompt: detailed system prompt for the AI assistant inside the demo | |
| - model: best Crusoe model for this demo: | |
| • "deepseek-ai/DeepSeek-R1" — reasoning, analysis, step-by-step thinking | |
| • "moonshotai/Kimi-K2-Instruct" — coding, long documents, structured output | |
| • "Qwen/Qwen3-235B-A22B" — general purpose, fast responses, multilingual | |
| - features: list of 2-4 key features to highlight | |
| CONDITIONAL (chatbot only): | |
| - chat_placeholder: placeholder text shown in the chat input box | |
| CONDITIONAL (comparison only): | |
| - model_a: first model ID (from the list above) | |
| - model_a_label: friendly display name for model A | |
| - model_b: second model ID | |
| - model_b_label: friendly display name for model B | |
| CONDITIONAL (dashboard only): | |
| - input_label: label for the main input text area | |
| - input_placeholder: placeholder text for that input | |
| CONDITIONAL (form_wizard only): | |
| - steps: list of 3-5 objects, each {"key": "snake_case_name", "question": "Question text?"} | |
| Template selection guide: | |
| - chatbot: conversational Q&A, support bots, advisors | |
| - comparison: show two models side-by-side on the same prompt | |
| - dashboard: analyze / summarize pasted text, data, or documents | |
| - form_wizard: multi-step intake flows that end with AI recommendations | |
| Output ONLY the JSON object. No markdown, no explanations.""" | |
| def parse_intent(prompt: str) -> dict: | |
| response = client.chat.completions.create( | |
| model=INTENT_MODEL, | |
| messages=[ | |
| {"role": "system", "content": SYSTEM_PROMPT}, | |
| {"role": "user", "content": f"Create a demo for: {prompt}"}, | |
| ], | |
| temperature=0.3, | |
| max_tokens=2000, | |
| ) | |
| content = response.choices[0].message.content or "" | |
| # Strip DeepSeek R1 chain-of-thought tags | |
| content = re.sub(r"<think>.*?</think>", "", content, flags=re.DOTALL).strip() | |
| # Extract the JSON block if surrounded by markdown fences | |
| fenced = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", content, re.DOTALL) | |
| if fenced: | |
| content = fenced.group(1) | |
| else: | |
| # Grab the outermost { ... } | |
| brace = re.search(r"\{.*\}", content, re.DOTALL) | |
| if brace: | |
| content = brace.group() | |
| intent = json.loads(content) | |
| return _apply_defaults(intent) | |
| def _apply_defaults(intent: dict) -> dict: | |
| intent.setdefault("template_type", "chatbot") | |
| intent.setdefault("title", "AI Demo") | |
| intent.setdefault("description", "Powered by Crusoe Managed Inference") | |
| intent.setdefault("system_prompt", "You are a helpful AI assistant.") | |
| intent.setdefault("model", "Qwen/Qwen3-235B-A22B") | |
| intent.setdefault("features", []) | |
| # Chatbot defaults | |
| intent.setdefault("chat_placeholder", "Ask me anything...") | |
| # Comparison defaults | |
| intent.setdefault("model_a", "deepseek-ai/DeepSeek-R1") | |
| intent.setdefault("model_a_label", "DeepSeek R1") | |
| intent.setdefault("model_b", "moonshotai/Kimi-K2-Instruct") | |
| intent.setdefault("model_b_label", "Kimi K2") | |
| # Dashboard defaults | |
| intent.setdefault("input_label", "Paste your content here:") | |
| intent.setdefault("input_placeholder", "Enter text to analyze...") | |
| # Form wizard defaults | |
| intent.setdefault("steps", [ | |
| {"key": "requirement", "question": "What is your main requirement?"}, | |
| {"key": "context", "question": "Can you describe your use case in more detail?"}, | |
| {"key": "constraints", "question": "Are there any constraints or preferences to keep in mind?"}, | |
| ]) | |
| return intent | |