Update app.py

app.py

@@ -1,10 +1,29 @@
+import os
 import streamlit as st
-from transformers import pipeline
+import openai
 
-# Initialize the text generation pipeline
-generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+# Read OpenAI API key from environment variable
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-print("Pipeline initialized.")
+if not openai.api_key:
+    st.error("No OpenAI API key found in the environment variable OPENAI_API_KEY")
+
+def generate_test_cases(topic, num_cases):
+    if topic:
+        prompt = f"Generate {num_cases} manual test cases for the topic: {topic}"
+        response = openai.Completion.create(
+            engine="text-davinci-002",
+            prompt=prompt,
+            max_tokens=150,
+            n=num_cases,
+            stop=None,
+            temperature=0.7,
+        )
+
+        test_cases = response.choices
+        return [f"Test Case {i+1}:\n{case.text.strip()}" for i, case in enumerate(test_cases)]
+    else:
+        return ["Please enter a topic/subject."]
 
 # Streamlit interface
 st.title("Manual Test Case Generator")
@@ -12,12 +31,6 @@ 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 topic:
-        prompt = f"Generate {num_cases} manual test cases for the topic: {topic}"
-        print(f"Prompt - {prompt}")
-        test_cases = generator(prompt, max_length=150, do_sample=True, num_return_sequences=num_cases)
-        for i, case in enumerate(test_cases):
-            st.write(f"Test Case {i+1}:")
-            st.write(case["generated_text"])
-    else:
-        st.error("Please enter a topic/subject.")
+    test_cases = generate_test_cases(topic, num_cases)
+    for case in test_cases:
+        st.write(case)