Update app.py

app.py

@@ -1,24 +1,24 @@
-from flask import Flask, request, jsonify
+import gradio as gr
 from transformers import pipeline
 
-app = Flask(__name__)
+# Load an open-source model for free-tier compatibility
+generator = pipeline("text-generation", model="google/flan-t5-base")
 
-# Load a model for text generation
-generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct")
-
-@app.route("/generate", methods=["POST"])
-def generate_test_cases():
-    data = request.json
-    user_story = data.get("user_story", "")
-
-    if not user_story:
-        return jsonify({"error": "User story is required"}), 400
-
-    # Generate test cases
-    prompt = f"Generate JSON test cases from this user story: {user_story}"
+def generate_test_cases(user_story):
+    prompt = f"Generate structured test cases in JSON format from this user story: {user_story}"
+    
     output = generator(prompt, max_length=300, do_sample=True)[0]["generated_text"]
 
-    return jsonify({"test_cases": output})
+    return output
+
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=generate_test_cases,
+    inputs=gr.Textbox(lines=5, placeholder="Enter your user story here..."),
+    outputs="text",
+    title="AI Test Case Generator",
+    description="Enter a user story and get structured test cases in JSON format.",
+)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    iface.launch()