Update app.py

app.py

@@ -1,41 +1,28 @@
 import streamlit as st
 from fpdf import FPDF
-from groq import Groq
-import os
 
-# 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()
+# Streamlit App UI
+st.title("Class Timetable Generator")
+st.write("Generate and manage class schedules effortlessly.")
 
-# Initialize the Groq client
-client = Groq(api_key=api_key)
+# Input fields for timetable generation
+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)")
 
-# Function to generate a timetable
+# Function to generate a simple timetable
 def generate_timetable(teachers, subjects, slots):
     """
-    Generate a timetable using the Groq API.
+    Generate a basic timetable by assigning teachers to subjects and slots.
     """
-    try:
-        prompt = f"""
-        Create a weekly timetable given the following data:
-        Teachers: {", ".join(teachers)}
-        Subjects: {", ".join(subjects)}
-        Available Slots: {", ".join(slots)}.
-        """
-        
-        chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="llama3-8b-8192",
-        )
-        
-        return chat_completion.choices[0].message.content
-
-    except Exception as e:
-        st.error(f"Error generating timetable: {e}")
-        return None
+    timetable = []
+    for i in range(min(len(teachers), len(subjects), len(slots))):
+        timetable.append(f"Slot: {slots[i]} | Subject: {subjects[i]} | Teacher: {teachers[i]}")
+    
+    if len(timetable) == 0:
+        timetable.append("No timetable could be generated. Ensure there are enough teachers, subjects, and slots.")
+    
+    return "\n".join(timetable)
 
 # Function to save timetable as PDF
 def save_timetable_as_pdf(timetable, filename="timetable.pdf"):
@@ -55,15 +42,7 @@ def save_timetable_as_pdf(timetable, filename="timetable.pdf"):
     pdf.output(filename)
     return filename
 
-# Streamlit App UI
-st.title("Class Timetable Generator")
-st.write("Generate and manage class schedules effortlessly.")
-
-# Input fields
-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)")
-
+# Generate timetable based on input
 if st.button("Generate Timetable"):
     if teachers_input and subjects_input and slots_input:
         teachers = [t.strip() for t in teachers_input.split(",")]
@@ -73,16 +52,13 @@ if st.button("Generate Timetable"):
         st.info("Generating timetable...")
         timetable = generate_timetable(teachers, subjects, slots)
         
-        if timetable:
-            st.success("Timetable generated successfully!")
-            st.text_area("Generated Timetable", timetable, height=300)
+        st.success("Timetable generated successfully!")
+        st.text_area("Generated Timetable", timetable, height=300)
 
-            # Option to download the timetable as a PDF
-            if st.button("Download PDF"):
-                pdf_file = save_timetable_as_pdf(timetable)
-                with open(pdf_file, "rb") as file:
-                    st.download_button("Download Timetable PDF", file, file_name="timetable.pdf")
-        else:
-            st.error("Failed to generate the timetable. Please check the inputs or API settings.")
+        # Option to download the timetable as a PDF
+        if st.button("Download PDF"):
+            pdf_file = save_timetable_as_pdf(timetable)
+            with open(pdf_file, "rb") as file:
+                st.download_button("Download Timetable PDF", file, file_name="timetable.pdf")
     else:
         st.warning("Please fill all fields to generate the timetable.")