Update streamlit_app.py

streamlit_app.py

@@ -1,15 +1,18 @@
 import streamlit as st
 from groq import Groq
+import os
 import docx2txt
 import PyPDF2
 
-# -----------------------------
+# Hugging Face Secret for API key
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+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(
@@ -20,12 +23,13 @@ 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
@@ -40,7 +44,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=10)
+num_testcases = st.number_input("Approximate number of test cases to generate", min_value=1, max_value=50, value=5)
 
 generate_btn = st.button("Generate Test Cases")
 
@@ -64,7 +68,7 @@ def read_file(file):
 # ------------------------------
 # Call Groq API to generate test cases
 # ------------------------------
-def call_groq_generate_testcases(text, num_testcases=10):
+def call_groq_generate_testcases(text, num_testcases=5):
     prompt = f"""
 You are a QA expert. For the following requirement, generate approximately {num_testcases} detailed test cases in the following clean format:
 
@@ -79,14 +83,14 @@ Steps:
 Expected Result: <Expected Result>
 ====================================================
 
-ONLY return the test cases in this clean format with lines separating each test case. Do not include anything else.
+ONLY return the test cases in this clean format with single line separators.
 Requirement:
 \"\"\"{text}\"\"\"
 """
     response = client.chat.completions.create(
         model="llama-3.1-8b-instant",
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=1500
+        messages=[{"role":"user", "content": prompt}],
+        max_tokens=2000
     )
     return response.choices[0].message.content
 
@@ -94,6 +98,7 @@ Requirement:
 # Generate test cases
 # ------------------------------
 if generate_btn:
+    # Get requirement text
     if input_method == "Upload File":
         if uploaded_file is not None:
             try:
@@ -118,10 +123,14 @@ if generate_btn:
         st.error(f"Error generating test cases: {e}")
         st.stop()
 
+    # Display test cases as card style
     st.subheader("🔸 Generated Test Cases")
-    with st.expander("View Test Cases", expanded=True):
-        st.code(testcases_output, language="text")
+    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)
 
+    # Copy + Download
     st.download_button(
         "Download Test Cases as TXT",
         testcases_output,