Update app.py

app.py

@@ -1,42 +1,61 @@
 import gradio as gr
 from coordinator.coordinator import Coordinator
-from coordinator.task_assigner import assign_tasks as llm_assign
+from coordinator.task_parser import parse_brief_with_reasoning
+from coordinator.task_assigner import assign_tasks_with_reasoning
+from frontend_agent.ui_generator import generate_react_component_llm
+from backend_agent.api_generator import generate_backend_code_llm
 
-# Initialize Coordinator
 coord = Coordinator()
 
-def process_brief(project_brief):
-    """
-    Takes the project brief and runs all steps automatically,
-    showing them sequentially in the output.
-    """
+def process_brief_with_reasoning(project_brief):
     output_text = ""
 
-    # Step 1: Parse tasks
-    tasks = coord.run(project_brief)  # returns dict {task: agent}, but only tasks for now
-    parsed_tasks = list(tasks.keys())
-    output_text += "### Step 1: Parsed Tasks\n"
-    for t in parsed_tasks:
+    # Step 1: Parsing Tasks with reasoning
+    output_text += "### Step 1: Parsing Tasks\n"
+    parsing_result = parse_brief_with_reasoning(project_brief)
+    output_text += "**LLM Reasoning:**\n"
+    output_text += parsing_result["reasoning"] + "\n\n"
+    output_text += "**Parsed Tasks:**\n"
+    for t in parsing_result["tasks"]:
         output_text += f"- {t}\n"
 
-    # Step 2: Assign tasks
-    assignments = llm_assign(parsed_tasks)
-    output_text += "\n### Step 2: Assigned Tasks\n"
-    for task, agent in assignments.items():
+    # Step 2: Assigning Tasks with reasoning
+    output_text += "\n### Step 2: Assigning Tasks\n"
+    assignment_result = assign_tasks_with_reasoning(parsing_result["tasks"])
+    output_text += "**LLM Reasoning:**\n"
+    output_text += assignment_result["reasoning"] + "\n\n"
+    output_text += "**Assigned Tasks:**\n"
+    for task, agent in assignment_result["assignments"].items():
         output_text += f"- {task} -> {agent}\n"
 
-    # Future steps (Frontend/Backend generation) can be added here
-    # output_text += "\n### Step 3: Frontend/Backend Code Generated\n..."
+    # Step 3: Code Generation
+    output_text += "\n### Step 3: Code Generation\n"
+    for task, agent in assignment_result["assignments"].items():
+        output_text += f"\n#### {task} ({agent})\n"
+        if "frontend" in agent.lower():
+            code = generate_react_component_llm(task)
+            output_text += f"\n```jsx\n{code}\n```\n"
+        elif "backend" in agent.lower():
+            code = generate_backend_code_llm(task)
+            output_text += f"\n```python\n{code}\n```\n"
 
     return output_text
 
 with gr.Blocks() as demo:
-    gr.Markdown("## AI Project Coordinator (Automatic Step-by-Step)")
+    gr.Markdown("## AI Project Coordinator - Step-by-Step Reasoning Log")
     project_brief_input = gr.Textbox(lines=3, placeholder="Enter your project brief here...")
     submit_button = gr.Button("Submit")
-    output_box = gr.Markdown()
-
-    submit_button.click(fn=process_brief, inputs=project_brief_input, outputs=output_box)
+    console_output = gr.Textbox(
+        label="Processing Log",
+        placeholder="LLM reasoning and outputs will appear here...",
+        lines=25
+    )
+
+    submit_button.click(
+        fn=process_brief_with_reasoning,
+        inputs=project_brief_input,
+        outputs=console_output
+    )
 
 if __name__ == "__main__":
     demo.launch()