added test_case and test_script

app.py

@@ -8,28 +8,70 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 if not openai.api_key:
     st.error("No OpenAI API key found in the environment variable OPENAI_API_KEY")
 
-conversation = [{"role": "system", "content": "You are a technical and a professional QA manager, working in a technological firm. You provide test cases for a scenario. "},]
+conversation_tc = [{"role": "system", "content": "You are a technical and a professional QA manager, working in a technological firm. You provide test cases for a scenario. "}]
+convo_py = [{"role": "system", "content": "You are a technical and a professional QA manager who specializes in Python Unittest, working in a technological firm. You should provide Python Unittest test scripts in for the test case provided"}]
+convo_java = [{"role": "system", "content": "You are a technical and a professional QA manager who specializes in Java JUnit, working in a technological firm. You should provide Java JUnit test scripts in for the test case provided"}]
+
 
 def generate_test_cases(topic, num_cases):
     if topic:
-        conversation.append({"role": "user", "content": f"Generate {num_cases} manual test cases for the topic: {topic}"})
+        conversation_tc.append({"role": "user", "content": f"Generate {num_cases} manual test cases for the topic: {topic}"})
         response = openai.ChatCompletion.create(
               model="gpt-3.5-turbo",
-              messages=conversation
+              messages=conversation_tc
           )
 
         test_cases = response["choices"][0]["message"]["content"]
-        # return [f"Test Case {i+1}:\n{case.text.strip()}" for i, case in enumerate(test_cases)]
         return test_cases
     else:
-        return ["Please enter a topic/subject."]
+        return "Please enter a topic/subject."
+
+def generate_test_scripts(language, framework, test_cases):
+    if language == "Python" and framework == "unittest":
+        return generate_python_unittest(test_cases)
+    elif language == "Java" and framework == "JUnit":
+        return generate_java_junit(test_cases)
+    else:
+        return "Unsupported language or framework."
+
+def generate_python_unittest(test_cases):
+    convo_py.append({"role": "user", "content": f"Here is a manual test case. {test_cases}"})
+    # prompt = f"Create a Python unittest test script for the following test cases:\n{test_cases}"
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=convo_py
+    )
+    script = response["choices"][0]["message"]["content"]
+    return script
+
+def generate_java_junit(test_cases):
+    convo_java.append({"role": "user", "content": f"Here is a manual test case. {test_cases}"})
+    # prompt = f"Create a Java JUnit test script for the following test cases:\n{test_cases}"
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=convo_java
+    )
+    script = response["choices"][0]["message"]["content"]
+    return script
 
 # Streamlit interface
 st.title("Manual Test Case Generator")
-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"):
-    test_cases = generate_test_cases(topic, num_cases)
-    # for case in test_cases:
-    st.write(test_cases)
+# Columns layout
+col1, col2 = st.beta_columns(2)
+
+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"):
+        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)