Update app.py

app.py

@@ -15,7 +15,6 @@ HF_API_KEY = os.getenv("HF_API_KEY")
 # ✅ Initialize Hugging Face API Client
 client = InferenceClient(token=HF_API_KEY)
 
-
 # ✅ Load Local Model with Device Optimization
 MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.2" 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -99,3 +98,20 @@ if st.button("Ask AI via API"):
 if st.button("Ask AI via Local Model"):
     ai_response = ask_mistral_local(user_query)
     st.write("🧠 **Mistral AI Suggests:**", ai_response)
+
+# ✅ Auto-Schedule Feature
+st.markdown("## 📅 Auto-Schedule Missing Timetable Slots")
+selected_file = st.selectbox("Choose a timetable file to auto-fill missing slots:", list(uploaded_files.keys()))
+
+if st.button("Auto-Schedule"):
+    result = auto_schedule(uploaded_files[selected_file])
+    st.write("✅", result)
+
+# ✅ Display Uploaded Timetables
+st.markdown("## 📜 View Uploaded Timetables")
+
+for name, file in uploaded_files.items():
+    if file:
+        df = pd.read_excel(file)
+        st.markdown(f"### {name}")
+        st.dataframe(df)