Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import pandas as pd
 import openpyxl
-import xlrd
 import docx
 from io import BytesIO
 from reportlab.pdfgen import canvas
@@ -43,15 +42,15 @@ if API_KEY:
         return pd.ExcelFile(file)
 
     # Function to edit Excel with full tools
-    def edit_excel(df):
-        st.write("### Excel Editor with Full Editing Tools")
-        
+    def edit_excel(df, file_key):
+        st.write(f"### Editing Excel: {file_key}")
+
         # Editable DataFrame
         edited_df = st.data_editor(df, num_rows="dynamic", use_container_width=True)
 
         # Apply Formatting Options
-        col_to_format = st.selectbox("Select Column to Format:", df.columns)
-        if st.button("Apply Bold Format"):
+        col_to_format = st.selectbox(f"Select Column to Format in {file_key}:", df.columns, key=f"col_{file_key}")
+        if st.button(f"Apply Bold Format - {file_key}", key=f"btn_{file_key}"):
             edited_df[col_to_format] = edited_df[col_to_format].apply(lambda x: f"**{x}**")
 
         return edited_df
@@ -68,37 +67,39 @@ if API_KEY:
         c.save()
         return output
 
-    # **📂 AI-Powered File Manager UI**
-    st.title("📂 AI-Powered File Manager with NLP & Excel Editor")
+    # 🏠 Streamlit UI
+    st.title("📂 RAG + TinyLlama NLP File Manager")
     st.write("Upload, edit, and manage `.docx`, `.xls` files with AI-powered NLP and full Excel tools.")
 
     # 📤 File Uploaders
-    uploaded_files = st.file_uploader("📄 Upload `.docx` or `.xlsx` Files", type=["docx", "xls", "xlsx"], accept_multiple_files=True)
+    uploaded_files = {}
+    for i in range(1, 5):
+        uploaded_files[f"File {i}"] = st.file_uploader(f"📄 Upload File {i}", type=["docx", "xls"])
 
     # ✏️ File Editors
     edited_docs = {}
     edited_excels = {}
 
-    for file in uploaded_files:
+    for file_key, file in uploaded_files.items():
         if file:
-            st.subheader(f"📝 Editing File: {file.name}")
+            st.subheader(f"📝 Editing {file_key}: {file.name}")
 
             if file.name.endswith(".docx"):
                 text = load_docx(file)
                 edited_text = st.text_area(f"Edit {file.name}:", text, height=300)
 
                 # AI Text Suggestion
-                if st.button(f"✨ Generate AI Suggestion for {file.name}"):
+                if st.button(f"✨ Generate AI Suggestion for {file.name}", key=f"suggest_{file_key}"):
                     suggestion = generate_suggestion(edited_text)
                     st.text_area("🤖 AI Suggestion:", suggestion, height=200)
 
                 edited_docs[file.name] = edited_text
 
-            elif file.name.endswith(".xls") or file.name.endswith(".xlsx"):
+            elif file.name.endswith(".xls"):
                 xls = load_excel(file)
-                sheet_name = st.selectbox(f"📊 Select Sheet in {file.name}:", xls.sheet_names)
+                sheet_name = st.selectbox(f"📊 Select Sheet in {file.name}:", xls.sheet_names, key=f"sheet_{file_key}")
                 df = pd.read_excel(xls, sheet_name=sheet_name)
-                edited_df = edit_excel(df)
+                edited_df = edit_excel(df, file_key)
                 edited_excels[file.name] = edited_df
 
     # 📥 Download Buttons
@@ -130,3 +131,23 @@ if API_KEY:
             file_name=file_name,
             mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
         )
+
+    # 🗨️ Chat with TinyLlama about Timetable Conflicts
+    st.subheader("🗓️ Timetable Conflict Checker")
+    query = st.text_input("Ask about conflicts in the uploaded Timetables:")
+
+    if query:
+        conflict_info = "No conflicts detected."  # Default response
+        for file_key, file in uploaded_files.items():
+            if file and file.name.endswith(".xls"):
+                xls = load_excel(file)
+                for sheet in xls.sheet_names:
+                    df = pd.read_excel(xls, sheet_name=sheet)
+                    if "Monday" in df.columns:  # Adjust this based on column headers
+                        monday_data = df["Monday"].dropna().astype(str).tolist()
+                        response = generate_suggestion(f"Analyze this schedule: {monday_data}. Find conflicts.")
+                        if "conflict" in response.lower():
+                            conflict_info = response
+
+        st.write("📢 **Conflict Analysis:**")
+        st.write(conflict_info)