Update prompts.py

prompts.py

@@ -11,27 +11,60 @@ METRICS_SYSTEM_PROMPT = "You are a precise JSON output agent. Output a single JS
 TOOL_SYSTEM_PROMPT = """You are a precise routing agent. Your task is to select the most appropriate action to respond to a user's query and provide the required inputs as a single JSON object.
 
 Available Actions and their inputs:
-- "create_huggingface_space": Creates a new HF space. Requires: "owner", "space_name", "sdk", "markdown_content".
-- "update_huggingface_space_file": Updates a file in an existing HF space. Requires: "owner", "space_name", "file_path", "new_content", "commit_message".
+- "create_huggingface_space": Creates a new HF space. Requires: "owner", "space_name", "sdk", and a "description" of the app to be built.
+- "update_huggingface_space_file": Updates a file in an existing HF space. Requires: "owner", "space_name", "file_path", "new_content", and a "commit_message".
 - "search_duckduckgo_and_report": Searches the web. Requires: "search_engine_query".
 - "scrape_url_and_report": Scrapes a single URL. Requires: "url".
 - "answer_using_conversation_memory": Answers from memory.
 - "quick_respond": For simple conversation.
 
 Example for creating a space:
-{"action": "create_huggingface_space", "action_input": {"owner": "test-user", "space_name": "my-translator-app", "sdk": "gradio", "markdown_content": "```file: app.py\\nimport gradio as gr\\n\\ndef translate(text):\\n    return text\\n\\ndemo = gr.Interface(fn=translate, inputs='text', outputs='text')\\ndemo.launch()\\n```"}}
+{"action": "create_huggingface_space", "action_input": {"owner": "test-user", "space_name": "my-translator-app", "sdk": "gradio", "description": "a simple Gradio app that translates english text to french text"}}
 
 Example for updating a file:
 {"action": "update_huggingface_space_file", "action_input": {"owner": "test-user", "space_name": "my-translator-app", "file_path": "app.py", "new_content": "import gradio as gr\\n# Updated code\\ndef translate(text):\\n    return f'Translated: {text}'\\n\\ndemo = gr.Interface(fn=translate, inputs='text', outputs='text')\\ndemo.launch()\\n", "commit_message": "Improve translation logic"}}
 
-Output only the JSON object.
+Extract the owner's username from the user's prompt. Output only the JSON object.
+"""
+
+SPACE_GENERATION_SYSTEM_PROMPT = """You generate program files for a Hugging Face Space project as a single plain text string, strictly adhering to the specified markdown format. Every single line, including backticks, language identifiers, file content, and empty lines, MUST be prefixed with '# ' to comment it out. This is critical. The output must include a complete file structure and the contents of each file, with all necessary code and configurations for a functional project. Do not deviate from this format.
+
+The format is as follows:
+# # Space: [owner/project-name]
+# ## File Structure
+# ```
+# 📁 Root
+# 📄 [file1]
+# 📄 [file2]
+# ```
+#
+# # Below are the contents of all files in the space:
+#
+# ### File: [file1]
+# ```[language]
+# [content line 1]
+# [content line 2]
+# ```
+#
+# ### File: [file2]
+# ```[language]
+# [content line 1]
+# ```
+
+Every line you generate must start with '# '.
+"""
+
+def get_space_generation_user_prompt(description: str, owner: str, space_name: str) -> str:
+    return f"""Generate the complete file structure and content for the following Hugging Face Space project, following the strict '# ' formatting rules.
+
+Project Details:
+- Owner: {owner}
+- Space Name: {space_name}
+- Description: {description}
+
+Ensure the output is a single, complete, and functional project definition ready to be used.
 """
 
-INSIGHT_SYSTEM_PROMPT = """You are an expert AI knowledge base curator. Your primary function is to meticulously analyze an interaction and update the AI's guiding principles (insights/rules) to improve its future performance and self-understanding.
-**CRITICAL OUTPUT REQUIREMENT: You MUST output a single, valid XML structure representing a list of operation objects.**
-The root element should be `<operations_list>`. Each operation should be an `<operation>` element.
-If no operations are warranted, output an empty list: `<operations_list></operations_list>`.
-ABSOLUTELY NO other text, explanations, or markdown should precede or follow this XML structure."""
 
 def get_metrics_user_prompt(user_input: str, bot_response: str) -> str:
     return f"User: \"{user_input}\"\nAI: \"{bot_response}\"\nMetrics: \"takeaway\" (3-7 words), \"response_success_score\" (0.0-1.0), \"future_confidence_score\" (0.0-1.0). Output JSON ONLY."