Update app.py

app.py

@@ -3,9 +3,14 @@ from fpdf import FPDF
 from groq import Groq
 import os
 
-# Set the Groq API key from Hugging Face Secrets (ensure to add it in your Space's settings)
-api_key = os.getenv("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
-")
+# Retrieve the API key securely
+# Ensure the "GROQ_API_KEY" is added in Hugging Face Secrets or as an environment variable
+api_key = os.getenv("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")
+if not api_key:
+    st.error("API Key not found. Please set the GROQ_API_KEY environment variable in your Space settings.")
+    st.stop()
+
+# Initialize the Groq client
 client = Groq(api_key=api_key)
 
 # Function to generate a timetable
@@ -55,12 +60,16 @@ st.title("Class Timetable Generator")
 st.write("Generate and manage class schedules effortlessly.")
 
 # Input fields
-teachers = st.text_input("Enter the names of teachers (comma-separated)").split(",")
-subjects = st.text_input("Enter the subjects (comma-separated)").split(",")
-slots = st.text_input("Enter the available slots (comma-separated)").split(",")
+teachers_input = st.text_input("Enter the names of teachers (comma-separated)")
+subjects_input = st.text_input("Enter the subjects (comma-separated)")
+slots_input = st.text_input("Enter the available slots (comma-separated)")
 
 if st.button("Generate Timetable"):
-    if teachers and subjects and slots:
+    if teachers_input and subjects_input and slots_input:
+        teachers = [t.strip() for t in teachers_input.split(",")]
+        subjects = [s.strip() for s in subjects_input.split(",")]
+        slots = [s.strip() for s in slots_input.split(",")]
+
         st.info("Generating timetable...")
         timetable = generate_timetable(teachers, subjects, slots)