Create app.py

app.py

@@ -0,0 +1,106 @@
+import gradio as gr
+from nodes.DemandUnderstand import DemandUnderstand
+from nodes.Coding import Coding
+from nodes.CodeComplete import CodeComplete
+from nodes.CodeReviewComment import CodeReviewComment
+from nodes.CodeReviewModification import CodeReviewModification
+from nodes.TestErrorSummary import TestErrorSummary
+from nodes.TestModification import TestModification
+
+def process_input(input_text):
+    # Initialize nodes
+    demand = DemandUnderstand()
+    coding = Coding()
+    code_complete = CodeComplete()
+    code_review_comment = CodeReviewComment()
+    code_review_mod = CodeReviewModification()
+    test_error_summary = TestErrorSummary()
+    test_mod = TestModification()
+
+    # Step 1: Demand Understanding
+    demand_result = demand.func_prompt_enhance(input_text)
+
+    # Step 2: Coding
+    coding_result = coding.func_coding(
+        assistant_role="Programmer",
+        gui=demand_result["product_vision"],
+        ideas="Innovative Features",
+        language=demand_result["language"],
+        modality=demand_result["modality"],
+        task=demand_result["task"]
+    )
+
+    # Step 3: Code Complete
+    code_complete_result = code_complete.func_coding(
+        assistant_role="Programmer",
+        unimplemented_file=coding_result["unimplemented_file"],
+        codes=coding_result["output"],
+        language=demand_result["language"],
+        modality=demand_result["modality"],
+        task=demand_result["task"]
+    )
+
+    # Step 4: Code Review Comment
+    review_comment = code_review_comment.generate_comments(
+        assistant_role="Code Reviewer",
+        codes=code_complete_result["output"],
+        ideas="Innovative Features",
+        language=demand_result["language"],
+        modality=demand_result["modality"],
+        task=demand_result["task"]
+    )
+
+    # Step 5: Code Review Modification
+    review_mod = code_review_mod.func_coding(
+        assistant_role="Code Reviewer",
+        comments=review_comment["output"],
+        codes=code_complete_result["output"],
+        ideas="Innovative Features",
+        language=demand_result["language"],
+        modality=demand_result["modality"],
+        task=demand_result["task"]
+    )
+
+    # Step 6: Test Error Summary
+    error_summary = test_error_summary.summarize_errors(
+        assistant_role="Programmer",
+        test_reports="Initial Test Report",
+        codes=review_mod["output"],
+        language=demand_result["language"]
+    )
+
+    # Step 7: Test Modification
+    test_mod_result = test_mod.func_coding(
+        assistant_role="Programmer",
+        error_summary=error_summary["output"],
+        test_reports="Initial Test Report",
+        codes=review_mod["output"],
+        language=demand_result["language"]
+    )
+
+    # Compile Results
+    final_output = {
+        "Demand Understanding": demand_result,
+        "Coding": coding_result,
+        "Code Complete": code_complete_result,
+        "Code Review Comment": review_comment,
+        "Code Review Modification": review_mod,
+        "Test Error Summary": error_summary,
+        "Test Modification": test_mod_result
+    }
+
+    return final_output
+
+# Define the Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# AI-Driven Software Company")
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(label="Enter your project requirements", lines=4)
+            submit = gr.Button("Generate")
+        with gr.Column():
+            output = gr.JSON(label="Output")
+    submit.click(process_input, inputs=input_text, outputs=output)
+
+if __name__ == "__main__":
+    demo.launch()