Update app.py

app.py

@@ -58,20 +58,22 @@ def generate_java_junit(test_cases):
 st.title("Manual Test Case Generator")
 
 # Columns layout
-col1, col2 = st.beta_columns(2)
+col1, col2 = st.columns(2)
+test_cases = None
 
 with col1:
     topic = st.text_input("Enter a topic or subject:")
     num_cases = st.number_input("Number of test cases:", min_value=1, max_value=10, value=1)
 
-    if st.button("Generate Test Cases"):
+    if st.button("Generate Test Case"):
         test_cases = generate_test_cases(topic, num_cases)
         st.write(test_cases)
 
 with col2:
-    language = st.selectbox("Select a language:", ["Python", "Java"])
-    framework = st.selectbox("Select a test framework:", ["unittest", "JUnit"])
-
-    if st.button("Generate Test Scripts"):
-        test_scripts = generate_test_scripts(language, framework, test_cases)
-        st.code(test_scripts, language)
+    if test_cases:
+        language = st.selectbox("Select a language:", ["Python", "Java"])
+        framework = st.selectbox("Select a test framework:", ["unittest", "JUnit"])
+    
+        if st.button("Generate Test Script"):
+            test_scripts = generate_test_scripts(language, framework, test_cases)
+            st.code(test_scripts, language)