Upload 2 files

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load a lightweight test-case generation model
+generator = pipeline("text2text-generation", model="Salesforce/codet5p-770m", max_length=512)
+
+def generate_tests(code, instruction):
+    prompt = f"Input:\n{code}\n\nInstruction:\n{instruction}"
+    result = generator(prompt)[0]["generated_text"]
+    return result
+
+gr.Interface(
+    fn=generate_tests,
+    inputs=[
+        gr.Textbox(label="Your Code", lines=10, placeholder="Paste your function here..."),
+        gr.Textbox(label="Instruction", placeholder="e.g., Generate unit tests using Python unittest.")
+    ],
+    outputs=gr.Code(language="python"),
+    title="🧪 Unit Test & Test Case Generator",
+    description="🔧 Paste your function and get auto-generated test cases using a free model from Hugging Face."
+).launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch