Update streamlit_app.py

streamlit_app.py

@@ -5,10 +5,10 @@ import docx2txt
 import PyPDF2
 
 # ------------------------------
-# Hugging Face Spaces: API Key from Secrets
+# Get API key from Hugging Face Secrets
 # ------------------------------
-# Set your API key in the Space's Secrets with the name "GROQ_API_KEY"
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+client = Groq(api_key=GROQ_API_KEY)
 
 # ------------------------------
 # Streamlit page config and theme
@@ -90,7 +90,7 @@ Requirement:
 """
     response = client.chat.completions.create(
         model="llama-3.1-8b-instant",
-        messages=[{"role":"user", "content": prompt}],
+        messages=[{"role": "user", "content": prompt}],
         max_tokens=1500
     )
     return response.choices[0].message.content
@@ -125,7 +125,7 @@ if generate_btn:
 
     st.subheader("🔸 Generated Test Cases")
     with st.expander("View Test Cases", expanded=True):
-        st.text(testcases_output)
+        st.code(testcases_output, language="text")
 
     st.download_button(
         "Download Test Cases as TXT",