Update app.py

app.py

@@ -1,43 +1,24 @@
-import streamlit as st
-import backend  # Importing backend functions
+from flask import Flask, request, jsonify
+from transformers import pipeline
 
-# Title
-st.title("Test Case Generator - QA User Story")
+app = Flask(__name__)
 
-# Input field for the user story
-user_story = st.text_area("Enter the QA User Story", "")
+# Load a model for text generation
+generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct")
 
-# Button to generate test cases
-if st.button("Generate Test Cases"):
-    if user_story:
-        with st.spinner("Generating test cases..."):
-            test_cases = backend.generate_testcases(user_story)
-        st.subheader("Generated Test Cases")
-        if test_cases:
-            for case in test_cases:
-                st.write(f"- **Test Case**: {case.get('test_case', 'N/A')}")
-            st.session_state.test_cases = test_cases  # Store in session state
-        else:
-            st.error("Failed to generate test cases. Try a clearer input.")
-    else:
-        st.error("Please enter a user story.")
+@app.route("/generate", methods=["POST"])
+def generate_test_cases():
+    data = request.json
+    user_story = data.get("user_story", "")
 
-# Export test cases
-st.sidebar.title("Export Test Cases")
-if st.sidebar.button("Export Test Cases as CSV"):
-    if 'test_cases' in st.session_state:
-        test_cases = st.session_state.test_cases
-        csv_content = backend.export_test_cases(test_cases, 'csv')
-        if csv_content:
-            st.sidebar.download_button(
-                label="Download CSV",
-                data=csv_content,
-                file_name="test_cases.csv",
-                mime="text/csv"
-            )
-    else:
-        st.sidebar.error("No test cases available.")
+    if not user_story:
+        return jsonify({"error": "User story is required"}), 400
 
-# Footer
-st.markdown("---")
-st.write("Built by testcore")
+    # Generate test cases
+    prompt = f"Generate JSON test cases from this user story: {user_story}"
+    output = generator(prompt, max_length=300, do_sample=True)[0]["generated_text"]
+
+    return jsonify({"test_cases": output})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)