Upload admin.py

admin.py

@@ -0,0 +1,171 @@
+# import streamlit as st
+# import os
+# from PIL import Image
+# import cv2
+# import pytesseract
+# from docx import Document
+# from pptx import Presentation
+# from io import BytesIO
+# import base64
+# from zipfile import ZipFile
+# import pandas as pd
+
+# # Set the path where uploaded files will be stored
+# UPLOAD_FOLDER = 'data/'
+# os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
+# # Function to handle login
+# def login():
+#     st.subheader("Login")
+#     username = st.text_input("Username")
+#     password = st.text_input("Password", type="password")
+#     if username == "admin" and password == "secret":
+#         st.success("You are logged in!")
+#         return True
+#     else:
+#         st.error("Invalid credentials.")
+#         return False
+
+# # Function to handle file uploads
+# def upload_files():
+#     st.subheader("Upload Files")
+#     uploaded_files = st.file_uploader("Choose files", type=["pdf", "docx", "pptx", "jpg", "png", "txt"], accept_multiple_files=True)
+#     for uploaded_file in uploaded_files:
+#         file_path = os.path.join(UPLOAD_FOLDER, uploaded_file.name)
+#         with open(file_path, "wb") as f:
+#             f.write(uploaded_file.getbuffer())
+#         st.success(f"{uploaded_file.name} uploaded successfully!")
+
+# # Main function to run the app
+# def main():
+#     st.title("Admin Panel")
+    
+#     # Check if user is logged in
+#     if not login():
+#         st.stop()
+    
+#     # Display uploaded files (optional)
+#     files = os.listdir(UPLOAD_FOLDER)
+#     st.write("Uploaded Files:")
+#     for file in files:
+#         st.write(file)
+    
+#     # Upload files
+#     upload_files()
+
+# if __name__ == "__main__":
+#     main()
+
+
+
+import streamlit as st
+import os
+from PIL import Image
+import cv2
+import pytesseract
+from docx import Document
+from pptx import Presentation
+from io import BytesIO
+import base64
+from zipfile import ZipFile
+import pandas as pd
+
+
+# Set the path where uploaded files will be stored
+UPLOAD_FOLDER = 'data/'
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
+# Function to handle login
+def login():
+    st.sidebar.subheader("Login")
+    username = st.sidebar.text_input("Username")
+    password = st.sidebar.text_input("Password", type="password")
+    if st.sidebar.button("Login"):
+        if username == "admin" and password == "secret":
+            st.session_state.logged_in = True  # Set login state to True
+            st.sidebar.success("You are logged in!")
+            st.rerun()  # Rerun the app to refresh the page
+        else:
+            st.sidebar.error("Invalid credentials.")
+
+# Function to handle logout
+def logout():
+    if st.sidebar.button("Logout"):
+        st.session_state.logged_in = False  # Reset login state
+        st.sidebar.success("You have been logged out!")
+        st.rerun()  # Rerun the app to refresh the page
+
+# Function to handle file uploads
+def upload_files():
+    st.subheader("Upload Files")
+    uploaded_files = st.file_uploader("Choose files", type=["pdf", "docx", "pptx", "jpg", "png", "txt"], accept_multiple_files=True)
+    if uploaded_files:
+        for uploaded_file in uploaded_files:
+            file_path = os.path.join(UPLOAD_FOLDER, uploaded_file.name)
+            with open(file_path, "wb") as f:
+                f.write(uploaded_file.getbuffer())
+            st.success(f"✅ {uploaded_file.name} uploaded successfully!")  # Green success alert
+        st.rerun()  # Rerun the app to refresh the file list
+
+# Function to display and manage uploaded files
+def manage_files():
+    st.subheader("Manage Files")
+    files = os.listdir(UPLOAD_FOLDER)
+    
+    if files:
+        # Create a DataFrame for all files
+        file_data = []
+        for file in files:
+            file_path = os.path.join(UPLOAD_FOLDER, file)
+            file_size = os.path.getsize(file_path) / 1024  # Size in KB
+            file_type = file.split(".")[-1].upper()  # Extract file extension
+            file_data.append({"File Name": file, "Type": file_type, "Size (KB)": round(file_size, 2)})
+        
+        df = pd.DataFrame(file_data)
+        
+        # Display the table
+        st.dataframe(df, use_container_width=True)
+        
+        # Add options to delete or download files
+        selected_file = st.selectbox("Select a file to manage", files)
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("Delete Selected File"):
+                os.remove(os.path.join(UPLOAD_FOLDER, selected_file))
+                st.error(f"❌ {selected_file} deleted successfully!")  # Red error alert
+                st.rerun()  # Rerun the app to refresh the file list
+        with col2:
+            with open(os.path.join(UPLOAD_FOLDER, selected_file), "rb") as f:
+                st.download_button(
+                    label="Download Selected File",
+                    data=f,
+                    file_name=selected_file,
+                    mime="application/octet-stream"
+                )
+    else:
+        st.info("No files uploaded yet.")
+
+# Main function to run the app
+def main():
+    st.title("Admin Panel")
+    
+    # Initialize session state for login
+    if "logged_in" not in st.session_state:
+        st.session_state.logged_in = False
+    
+    # Check if user is logged in
+    if not st.session_state.logged_in:
+        login()
+        st.stop()  # Stop the app if not logged in
+    
+    # Logout button
+    logout()
+    
+    # Upload files
+    upload_files()
+    
+    # Manage files (table for all files)
+    manage_files()
+
+if __name__ == "__main__":
+    main()