Update app.py

app.py

@@ -266,6 +266,25 @@ def display_bookmarks(bookmarks_list):
     logger.info("HTML display generated")
     return cards
 
+# Function to handle sending messages in chat
+def send_message(user_message, chat_history, state_bookmarks):
+    if not user_message:
+        return chat_history, chat_history
+
+    # Append user message to chat history
+    chat_history = chat_history + [(user_message, None)]
+
+    # Generate chatbot response
+    try:
+        response = chatbot_response(user_message, state_bookmarks)
+    except Exception as e:
+        response = f"⚠️ Error: {str(e)}"
+
+    # Append assistant response to chat history
+    chat_history[-1] = (user_message, response)
+
+    return chat_history, chat_history
+
 # Process the uploaded file
 def process_uploaded_file(file, state_bookmarks):
     logger.info("Processing uploaded file")
@@ -451,46 +470,47 @@ def build_app():
     try:
         logger.info("Building Gradio app")
         with gr.Blocks(theme=gr.themes.Default(), css="app.css") as demo:
-            # Shared state to store bookmarks
+            # Shared states
             state_bookmarks = gr.State([])
+            chat_history = gr.State([])
 
             # General Overview
             gr.Markdown("""
-            # 📚 SmartMarks - AI Browser Bookmarks Manager
+# 📚 SmartMarks - AI Browser Bookmarks Manager
 
-            Welcome to **SmartMarks**, your intelligent assistant for managing browser bookmarks. SmartMarks leverages AI to help you organize, search, and interact with your bookmarks seamlessly. Whether you're looking to categorize your links, retrieve information quickly, or maintain an updated list, SmartMarks has you covered.
+Welcome to **SmartMarks**, your intelligent assistant for managing browser bookmarks. SmartMarks leverages AI to help you organize, search, and interact with your bookmarks seamlessly. Whether you're looking to categorize your links, retrieve information quickly, or maintain an updated list, SmartMarks has you covered.
 
-            ---
+---
 
-            ## 🚀 **How to Use SmartMarks**
+## 🚀 **How to Use SmartMarks**
 
-            SmartMarks is divided into three main sections:
+SmartMarks is divided into three main sections:
 
-            1. **📂 Upload and Process Bookmarks:** Import your existing bookmarks and let SmartMarks analyze and categorize them for you.
-            2. **💬 Chat with Bookmarks:** Interact with your bookmarks using natural language queries to find relevant links effortlessly.
-            3. **🛠️ Manage Bookmarks:** View, edit, delete, and export your bookmarks with ease.
+1. **📂 Upload and Process Bookmarks:** Import your existing bookmarks and let SmartMarks analyze and categorize them for you.
+2. **💬 Chat with Bookmarks:** Interact with your bookmarks using natural language queries to find relevant links effortlessly.
+3. **🛠️ Manage Bookmarks:** View, edit, delete, and export your bookmarks with ease.
 
-            Navigate through the tabs to explore each feature in detail.
-            """)
+Navigate through the tabs to explore each feature in detail.
+""")
 
             # Upload and Process Bookmarks Tab
             with gr.Tab("Upload and Process Bookmarks"):
                 gr.Markdown("""
-                ## 📂 **Upload and Process Bookmarks**
+## 📂 **Upload and Process Bookmarks**
 
-                ### 📝 **Steps to Upload and Process:**
+### 📝 **Steps to Upload and Process:**
 
-                1. **🔽 Upload Bookmarks File:**
-                   - Click on the **"📁 Upload Bookmarks HTML File"** button.
-                   - Select your browser's exported bookmarks HTML file from your device.
+1. **🔽 Upload Bookmarks File:**
+   - Click on the **"📁 Upload Bookmarks HTML File"** button.
+   - Select your browser's exported bookmarks HTML file from your device.
 
-                2. **⚙️ Process Bookmarks:**
-                   - After uploading, click on the **"⚙️ Process Bookmarks"** button.
-                   - SmartMarks will parse your bookmarks, fetch additional information, generate summaries, and categorize each link based on predefined categories.
+2. **⚙️ Process Bookmarks:**
+   - After uploading, click on the **"⚙️ Process Bookmarks"** button.
+   - SmartMarks will parse your bookmarks, fetch additional information, generate summaries, and categorize each link based on predefined categories.
 
-                3. **📄 View Processed Bookmarks:**
-                   - Once processing is complete, your bookmarks will be displayed in an organized and visually appealing format below.
-                """)
+3. **📄 View Processed Bookmarks:**
+   - Once processing is complete, your bookmarks will be displayed in an organized and visually appealing format below.
+""")
 
                 upload = gr.File(label="📁 Upload Bookmarks HTML File", type='binary')
                 process_button = gr.Button("⚙️ Process Bookmarks")
@@ -506,55 +526,72 @@ def build_app():
             # Chat with Bookmarks Tab
             with gr.Tab("Chat with Bookmarks"):
                 gr.Markdown("""
-                ## 💬 **Chat with Bookmarks**
+## 💬 **Chat with Bookmarks**
 
-                ### 🤖 **How to Interact:**
+### 🤖 **How to Interact:**
 
-                1. **✍️ Enter Your Query:**
-                   - In the **"✍️ Ask about your bookmarks"** textbox, type your question or keyword related to your bookmarks. For example, "Do I have any bookmarks about GenerativeAI?"
+1. **✍️ Enter Your Query:**
+   - In the **"✍️ Ask about your bookmarks"** textbox, type your question or keyword related to your bookmarks. For example, "Do I have any bookmarks about GenerativeAI?"
 
-                2. **📨 Submit Your Query:**
-                   - Click the **"📨 Send"** button to submit your query.
+2. **📨 Submit Your Query:**
+   - You can either press the **"📨 Send"** button or simply press the **Enter** key to submit your query.
 
-                3. **📈 Receive AI-Driven Responses:**
-                   - SmartMarks will analyze your query and provide relevant bookmarks that match your request, making it easier to find specific links without manual searching.
-                """)
+3. **📈 Receive AI-Driven Responses:**
+   - SmartMarks will analyze your query and provide relevant bookmarks that match your request, making it easier to find specific links without manual searching.
 
-                user_input = gr.Textbox(label="✍️ Ask about your bookmarks", placeholder="e.g., Do I have any bookmarks about GenerativeAI?")
-                chat_output = gr.Textbox(label="💬 Chatbot Response", interactive=False)
-                chat_button = gr.Button("📨 Send")
+4. **🗂️ View Chat History:**
+   - All your queries and the corresponding AI responses are displayed in the chat history for your reference.
+""")
+
+                with gr.Row():
+                    chat_history_display = gr.Chatbot(label="🗨️ Chat History")
+                    chat_input = gr.Textbox(
+                        label="✍️ Ask about your bookmarks",
+                        placeholder="e.g., Do I have any bookmarks about GenerativeAI?",
+                        lines=1,
+                        interactive=True
+                    )
+                    chat_button = gr.Button("📨 Send")
+
+                # When user presses Enter in chat_input
+                chat_input.submit(
+                    send_message,
+                    inputs=[chat_input, chat_history, state_bookmarks],
+                    outputs=[chat_history_display, chat_history_display]
+                )
 
+                # When user clicks Send button
                 chat_button.click(
-                    chatbot_response,
-                    inputs=[user_input, state_bookmarks],
-                    outputs=chat_output
+                    send_message,
+                    inputs=[chat_input, chat_history, state_bookmarks],
+                    outputs=[chat_history_display, chat_history_display]
                 )
 
             # Manage Bookmarks Tab
             with gr.Tab("Manage Bookmarks"):
                 gr.Markdown("""
-                ## 🛠️ **Manage Bookmarks**
+## 🛠️ **Manage Bookmarks**
 
-                ### 🗂️ **Features:**
+### 🗂️ **Features:**
 
-                1. **👁️ View Bookmarks:**
-                   - All your processed bookmarks are displayed here with their respective categories and summaries.
+1. **👁️ View Bookmarks:**
+   - All your processed bookmarks are displayed here with their respective categories and summaries.
 
-                2. **✅ Select Bookmarks:**
-                   - Use the checkboxes next to each bookmark to select one, multiple, or all bookmarks you wish to manage.
+2. **✅ Select Bookmarks:**
+   - Use the checkboxes next to each bookmark to select one, multiple, or all bookmarks you wish to manage.
 
-                3. **🗑️ Delete Selected Bookmarks:**
-                   - After selecting the desired bookmarks, click the **"🗑️ Delete Selected Bookmarks"** button to remove them from your list.
+3. **🗑️ Delete Selected Bookmarks:**
+   - After selecting the desired bookmarks, click the **"🗑️ Delete Selected Bookmarks"** button to remove them from your list.
 
-                4. **✏️ Edit Categories:**
-                   - Select the bookmarks you want to re-categorize.
-                   - Choose a new category from the dropdown menu labeled **"🆕 New Category"**.
-                   - Click the **"✏️ Edit Category of Selected Bookmarks"** button to update their categories.
+4. **✏️ Edit Categories:**
+   - Select the bookmarks you want to re-categorize.
+   - Choose a new category from the dropdown menu labeled **"🆕 New Category"**.
+   - Click the **"✏️ Edit Category of Selected Bookmarks"** button to update their categories.
 
-                5. **💾 Export Bookmarks:**
-                   - Click the **"💾 Export Bookmarks"** button to download your updated bookmarks as an HTML file.
-                   - This file can be uploaded back to your browser to reflect the changes made within SmartMarks.
-                """)
+5. **💾 Export Bookmarks:**
+   - Click the **"💾 Export Bookmarks"** button to download your updated bookmarks as an HTML file.
+   - This file can be uploaded back to your browser to reflect the changes made within SmartMarks.
+""")
 
                 manage_output = gr.Textbox(label="🔄 Manage Output", interactive=False)
                 bookmark_selector = gr.CheckboxGroup(label="✅ Select Bookmarks", choices=[])
@@ -577,7 +614,8 @@ def build_app():
                     else:
                         return [], ""
 
-                refresh_button.click(
+                # Link process_uploaded_file to update Manage Bookmarks Tab
+                process_button.click(
                     refresh_manage_tab,
                     inputs=[state_bookmarks],
                     outputs=[bookmark_selector, bookmark_display_manage]
@@ -602,8 +640,14 @@ def build_app():
                     outputs=download_link
                 )
 
-            logger.info("Launching Gradio app")
-            demo.launch(debug=True)
+                refresh_button.click(
+                    refresh_manage_tab,
+                    inputs=[state_bookmarks],
+                    outputs=[bookmark_selector, bookmark_display_manage]
+                )
+
+        logger.info("Launching Gradio app")
+        demo.launch(debug=True)
     except Exception as e:
         logger.error(f"Error building the app: {e}")
         print(f"Error building the app: {e}")