Update app.py

app.py

@@ -13,27 +13,14 @@ import json
 import pandas as pd
 import altair as alt
 from datetime import date
-import pickle
 
-# Constants and API Keys
-API_KEY_YOUTUBE = "AIzaSyA20DXMC3HeqHs9sOMQUQ041wEkgsoFXb4"
-HF_API_KEY = os.getenv("KEY")
-HF_MISTRAL_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
-DATA_FILE = "shortcuts_data.pkl"
-AUTHOR_PASSWORD = os.getenv("APP_PASSWORD")
-
-# Initialize services
+# Set up YouTube API
+API_KEY_YOUTUBE = "AIzaSyA20DXMC3HeqHs9sOMQUQ041wEkgsoFXb4"  # Replace with your YouTube Data API v3 key
 youtube = build('youtube', 'v3', developerKey=API_KEY_YOUTUBE)
 
-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)
+# Hugging Face Setup
+HF_API_KEY = os.getenv("KEY")
+HF_MISTRAL_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
 
 def chat_with_mistral_hf(prompt):
     if not HF_API_KEY:
@@ -167,92 +154,102 @@ def display_news(articles):
     else:
         st.error("No news articles found.")
 
-
-# All other functions remain the same (search_youtube, voice_search, get_wikipedia_summary, etc.)
-
 def main():
     st.set_page_config(page_title="Multi-Search Application", layout="wide")
 
-    # Initialize session states
+    # Initialize chat history if not present
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
-    if "shortcuts" not in st.session_state:
-        st.session_state.shortcuts = load_shortcuts()
 
-    # Sidebar setup
+    # Sidebar options
     st.sidebar.title("Options")
-    search_type = st.sidebar.radio("Select Search Type", 
-        ("Wikipedia", "Google", "YouTube", "News", "Chat", "Shortcuts"))
+    search_type = st.sidebar.radio("Select Search Type", ("Wikipedia", "Google", "YouTube", "News", "Chat"))
     
-    # Voice search and chat title
     if st.sidebar.button("Voice Search"):
         query = voice_search()
         if query:
             st.session_state.query = query
     else:
         st.session_state.query = ""
-    
+
+    # Chat-specific sidebar settings
     chat_title = "Temporary Chat"
     if search_type == "Chat":
         chat_title = st.sidebar.text_input("Rename Chat:", "Temporary Chat")
 
-    # Shortcuts management in sidebar
-    if search_type == "Shortcuts":
-        st.sidebar.header("Manage Shortcuts")
-        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, 
-                            AUTHOR_PASSWORD
-                        ))
-                        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.")
-
-    # Main content area
-    if search_type == "Shortcuts":
-        st.header("Educational Shortcuts")
-        with st.expander("Click to View Shortcuts", expanded=True):
-            if st.session_state.shortcuts:
-                cols = st.columns(4)
-                for i, (name, link, icon_data, user_password, _) in enumerate(st.session_state.shortcuts):
-                    with cols[i % 4]:
-                        st.image(icon_data, width=100)
-                        st.markdown(f"[{name}]({link})")
-                        password_input_delete = st.text_input(
-                            "Enter Password to Delete", 
-                            type="password", 
-                            key=f"delete_{i}"
-                        )
-                        if st.button(f"Delete {name}", key=f"delbtn_{i}"):
-                            if password_input_delete in [AUTHOR_PASSWORD, user_password]:
-                                del st.session_state.shortcuts[i]
-                                save_shortcuts(st.session_state.shortcuts)
-                                st.rerun()
-                            else:
-                                st.warning("Incorrect password")
-            else:
-                st.write("No shortcuts added yet.")
-        st.markdown("**Note:** Add educational web links for student learning purposes.")
+    # Last seen timestamp
+    st.sidebar.write(f"Last seen: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+
+    if search_type == "Wikipedia":
+        lang_map = {"English": "en", "Spanish": "es", "Chinese": "zh", "Hindi": "hi", "Telugu": "te"}
+        selected_lang = st.sidebar.selectbox("Wikipedia Language", list(lang_map.keys()))
+        summary_levels = ["Brief", "Detailed", "Bullet Points"]
+        summary_level = st.sidebar.selectbox("Summarization Level", summary_levels)
+        char_limit = st.sidebar.slider("Character Limit", min_value=100, max_value=2000, value=500, step=100)
+
+        st.title("Wikipedia Summary & Text-to-Speech")
+        query = st.text_input("Enter a topic to search on Wikipedia:", value=st.session_state.query)
+
+        if query:
+            lang_code = lang_map[selected_lang]
+            summary = get_wikipedia_summary(query, lang_code, char_limit, summary_level)
+            st.markdown(f"### Summary for: {query}")
+            st.write(summary)
+
+            tts_filename = f"{query}_speech.mp3"
+            if st.button("Play Text-to-Speech"):
+                text_to_speech(summary, tts_filename, lang=lang_code)
+                st.audio(tts_filename, format="audio/mp3")
+
+        st.write("---")
+        st.write("### Footer")
+        st.write("This is a Wikipedia search section. You can find detailed information and summaries here.")
+
+    elif search_type == "Google":
+        st.title("Google Search")
+        query = st.text_input("Enter a search query for Google:", value=st.session_state.query)
+        if query and st.button("Search"):
+            results = google_search("AIzaSyBvnTpjwspsYBMlHN4nMEvybEmZL8mwAQ4", "464947c4e602c4ee8", query)
+            display_google_results(results)
+
+        st.write("---")
+        st.write("### Footer")
+        st.write("This is a Google search section. Use it to find websites and online resources.")
+
+    elif search_type == "YouTube":
+        st.title("YouTube Search")
+        query = st.text_input("Enter a topic to search on YouTube:", value=st.session_state.query)
+        if query and st.button("Search"):
+            videos = search_youtube(query)
+            if videos:
+                for video in videos:
+                    st.write(f"[{video['title']}]({video['url']})")
+                    st.image(video['thumbnail'])
+                    st.video(video['url'])
+                    st.write("---")
+
+        st.write("---")
+        st.write("### Footer")
+        st.write("This is a YouTube search section. Watch videos directly from your search results.")
+
+    elif search_type == "News":
+        st.subheader("Select Date Range")
+        start_date = st.date_input("From", datetime.date.today() - datetime.timedelta(days=7))
+        end_date = st.date_input("To", datetime.date.today())
+        
+        st.title("News Search")
+        query = st.text_input("Enter a news topic to search:", value=st.session_state.query)
+        if query and st.button("Search"):
+            articles = search_news(query, start_date, end_date)
+            display_news(articles)
+
+        st.write("---")
+        st.write("### Footer")
+        st.write("This is a news search section. Find the latest news articles here.")
 
     elif search_type == "Chat":
         st.title(chat_title)
+        
         for chat in st.session_state.chat_history:
             with st.chat_message(chat["role"]):
                 st.write(chat["content"])
@@ -263,17 +260,18 @@ def main():
             if user_input.strip():
                 with st.spinner("Generating response..."):
                     response = chat_with_mistral_hf(user_input)
+                    
                     st.session_state.chat_history.append({"role": "user", "content": user_input})
                     st.session_state.chat_history.append({"role": "assistant", "content": response})
                     st.rerun()
             else:
                 st.warning("Please enter a prompt before clicking Generate Response.")
 
-    # Other search types (Wikipedia, Google, YouTube, News) remain unchanged
-
-    # PDF saving functionality
+    # Save chat history as PDF
     if st.button("Save Chat History as PDF"):
-        chat_history = [(msg['content'], "") for msg in st.session_state.chat_history if msg['role'] == 'user']
+        chat_history = []
+        if st.session_state.query:
+            chat_history.append((st.session_state.query, "Your response here"))
         file_name = st.text_input("Enter filename for PDF:", "chat_history.pdf")
         pdf_file = save_chat_history_as_pdf(chat_history, file_name)
         st.success(f"Chat history saved as PDF: {pdf_file}")