Update tools/orchestrator.py

tools/orchestrator.py

@@ -68,10 +68,32 @@ def orchestrate_and_respond(user_input: str, provider_name: str, model_display_n
     context_str = None
 
     if action_type == "create_huggingface_space":
-        yield "status", "[Tool: Creating Space...]"
-        params = ["owner", "space_name", "sdk", "markdown_content"]
+        params = ["owner", "space_name", "sdk", "description"]
         if all(p in action_input for p in params):
-            result = create_huggingface_space(**action_input)
+            yield "status", "[Tool: Generating space content...]"
+            description = action_input["description"]
+            owner = action_input["owner"]
+            space_name = action_input["space_name"]
+            sdk = action_input["sdk"]
+
+            space_gen_user_prompt = prompts.get_space_generation_user_prompt(description, owner, space_name)
+            space_gen_messages = [
+                {"role": "system", "content": prompts.SPACE_GENERATION_SYSTEM_PROMPT},
+                {"role": "user", "content": space_gen_user_prompt}
+            ]
+            
+            markdown_content = "".join(list(call_model_stream(
+                provider=provider_name, model_display_name=model_display_name,
+                messages=space_gen_messages, api_key_override=ui_api_key_override,
+                temperature=0.1, max_tokens=4096
+            )))
+            
+            yield "status", "[Tool: Creating Space with generated content...]"
+            
+            result = create_huggingface_space(
+                owner=owner, space_name=space_name, sdk=sdk,
+                markdown_content=markdown_content.strip()
+            )
             context_str = f"Tool Result (Create Space): {result.get('result') or result.get('error', 'Unknown outcome from tool.')}"
         else:
             context_str = "Tool Failed: Missing required parameters for create_huggingface_space. Required: " + ", ".join(params)