Update streamlit_app.py

streamlit_app.py

@@ -4,15 +4,22 @@ import os
 import docx2txt
 import PyPDF2
 
-# Hugging Face Secret for API key
+# ------------------------------
+# Load API key from Hugging Face Secret
+# ------------------------------
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+
+if not GROQ_API_KEY:
+    st.error("⚠️ GROQ_API_KEY secret not found. Please add it in Hugging Face Space Settings → Secrets.")
+    st.stop()
+
 client = Groq(api_key=GROQ_API_KEY)
 
 # ------------------------------
 # Streamlit page config and theme
 # ------------------------------
 st.set_page_config(page_title="🧪 TestCase Studio", layout="wide")
-st.title("🧪 TestCase Studio")
+st.title("TestCase Studio")
 
 # Custom CSS for clean theme
 st.markdown(
@@ -23,13 +30,12 @@ st.markdown(
     .stTextArea>div>div>textarea {font-size: 14px;}
     .stFileUploader>div>div>label {font-weight: bold;}
     .stExpanderHeader {font-weight: bold; font-size: 16px; color: #00796b;}
-    .testcase-box {background-color: #ffffff; padding: 10px; margin-bottom: 5px; border-radius: 8px; box-shadow: 0 0 5px rgba(0,0,0,0.1);}
     </style>
     """,
     unsafe_allow_html=True
 )
 
-st.write("Choose **one input method** to provide requirements:")
+st.write("Choose one input method to provide requirements:")
 
 # ------------------------------
 # Input method selection
@@ -44,7 +50,7 @@ if input_method == "Upload File":
 else:
     manual_text = st.text_area("Paste requirement text here", height=200)
 
-num_testcases = st.number_input("Approximate number of test cases to generate", min_value=1, max_value=50, value=5)
+num_testcases = st.number_input("Approximate number of test cases to generate", min_value=1, max_value=50, value=10)
 
 generate_btn = st.button("Generate Test Cases")
 
@@ -68,7 +74,7 @@ def read_file(file):
 # ------------------------------
 # Call Groq API to generate test cases
 # ------------------------------
-def call_groq_generate_testcases(text, num_testcases=5):
+def call_groq_generate_testcases(text, num_testcases=10):
     prompt = f"""
 You are a QA expert. For the following requirement, generate approximately {num_testcases} detailed test cases in the following clean format:
 
@@ -83,14 +89,14 @@ Steps:
 Expected Result: <Expected Result>
 ====================================================
 
-ONLY return the test cases in this clean format with single line separators.
+ONLY return the test cases in this clean format with lines separating each test case. Do not include anything else.
 Requirement:
 \"\"\"{text}\"\"\"
 """
     response = client.chat.completions.create(
         model="llama-3.1-8b-instant",
         messages=[{"role":"user", "content": prompt}],
-        max_tokens=2000
+        max_tokens=1500
     )
     return response.choices[0].message.content
 
@@ -98,7 +104,6 @@ Requirement:
 # Generate test cases
 # ------------------------------
 if generate_btn:
-    # Get requirement text
     if input_method == "Upload File":
         if uploaded_file is not None:
             try:
@@ -123,14 +128,10 @@ if generate_btn:
         st.error(f"Error generating test cases: {e}")
         st.stop()
 
-    # Display test cases as card style
     st.subheader("🔸 Generated Test Cases")
-    for block in testcases_output.split("===================================================="):
-        block = block.strip()
-        if block:
-            st.markdown(f'<div class="testcase-box"><pre>{block}</pre></div>', unsafe_allow_html=True)
+    with st.expander("View Test Cases", expanded=True):
+        st.code(testcases_output, language="text")  # Shows output in a neat code block
 
-    # Copy + Download
     st.download_button(
         "Download Test Cases as TXT",
         testcases_output,