Refactor SecureCode AI and implement unit test generator

2025-10-27 16:14:25 +03:00
parent 0f74c215df
commit b3b1293d6c
28 changed files with 94 additions and 1373 deletions
--- a/week4/community-contributions/salah/unit-tests-generator/app.py
+++ b/week4/community-contributions/salah/unit-tests-generator/app.py
@@ -0,0 +1,35 @@
+import gradio as gr
+from test_generator import generate_tests
+
+def create_interface():
+    with gr.Blocks(title="Unit Test Generator") as ui:
+        gr.Markdown("# Unit Test Generator")
+        gr.Markdown("Paste your Python code and get AI-generated unit tests")
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                code_input = gr.Code(
+                    label="Your Code",
+                    language="python",
+                    lines=15
+                )
+                generate_btn = gr.Button("Generate Tests", variant="primary")
+
+            with gr.Column(scale=1):
+                tests_output = gr.Textbox(
+                    label="Generated Tests",
+                    lines=15,
+                    interactive=False
+                )
+
+        generate_btn.click(
+            fn=generate_tests,
+            inputs=[code_input],
+            outputs=[tests_output]
+        )
+
+    return ui
+
+def launch():
+    ui = create_interface()
+    ui.launch(server_name="localhost", server_port=7860)