Update app.py

app.py

@@ -13,6 +13,7 @@ import json
 import pandas as pd
 import altair as alt
 from datetime import date
+import pickle
 
 # Set up YouTube API
 API_KEY_YOUTUBE = os.getenv("YT")  # Replace with your YouTube Data API v3 key
@@ -154,9 +155,52 @@ def display_news(articles):
     else:
         st.error("No news articles found.")
 
+# File to store shortcut data
+DATA_FILE = "shortcuts_data.pkl"
+
+def load_shortcuts():
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, "rb") as f:
+            return pickle.load(f)
+    return []
+
+def save_shortcuts(shortcuts):
+    with open(DATA_FILE, "wb") as f:
+        pickle.dump(shortcuts, f)
+
+if "shortcuts" not in st.session_state:
+    st.session_state["shortcuts"] = load_shortcuts()
+
+
 def main():
     st.set_page_config(page_title="Multi-Search Application", layout="wide")
 
+        # **Show Saved Shortcuts**
+    st.header("My Shortcuts")
+    with st.expander("Click to View Shortcuts"):
+        if st.session_state.shortcuts:
+            num_columns = 4
+            cols = st.columns(num_columns)
+            for i, (name, link, icon_data, user_password) in enumerate(st.session_state.shortcuts):
+                col = cols[i % num_columns]
+                with col:
+                    st.image(icon_data, width=100)
+                    st.markdown(f"[{name}]({link})", unsafe_allow_html=True)
+
+                    password_input_delete = st.text_input("Enter Password to Delete", type="password", key=f"delete_password_{i}")
+
+                    if password_input_delete == user_password:  
+                        if st.button(f"Delete {name}", key=f"delete_{i}"):
+                            st.session_state.shortcuts.pop(i)
+                            save_shortcuts(st.session_state.shortcuts)
+                            st.experimental_rerun()
+                    elif password_input_delete:
+                        st.warning("Incorrect password. You cannot delete this shortcut.")
+        else:
+            st.write("No shortcuts added yet.")
+
+    st.markdown("Note: Add best web links for student education purposes.")
+
     # Initialize chat history if not present
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
@@ -164,6 +208,30 @@ def main():
     # Sidebar options
     st.sidebar.title("Options")
     search_type = st.sidebar.radio("Select Search Type", ("Wikipedia", "Google", "YouTube", "News", "Chat"))
+
+    # **Add this block after st.sidebar.title("Options")**
+    st.sidebar.header("Only Add Best Web Links")
+    name = st.sidebar.text_input("Shortcut Name")
+    link = st.sidebar.text_input("Website Link (https://...)")
+    icon_file = st.sidebar.file_uploader("Upload an Icon", type=["png", "jpg", "jpeg"])
+    password_input_add = st.sidebar.text_input("Enter Password to Add Shortcut", type="password")
+
+    if st.sidebar.button("Add Shortcut"):
+        if password_input_add: 
+            if name and link and icon_file:
+                existing_links = [shortcut[1] for shortcut in st.session_state.shortcuts]
+                if link in existing_links:
+                    st.sidebar.warning("This website already exists.")
+                else:
+                    st.session_state.shortcuts.append((name, link, icon_file.getvalue(), password_input_add))
+                    save_shortcuts(st.session_state.shortcuts)
+                    st.sidebar.success(f"Shortcut '{name}' added!")
+            else:
+                st.sidebar.warning("Please enter all details including an icon.")
+        else:
+            st.sidebar.warning("Please enter a password to add the shortcut.")
+
+            
     
     if st.sidebar.button("Voice Search"):
         query = voice_search()