Update app.py

app.py

@@ -557,11 +557,11 @@ def edit_selected_bookmarks_category(selected_indices, new_category):
 
 def export_bookmarks():
     """
-    Export bookmarks to HTML file.
+    Export bookmarks to an HTML file.
     """
     if not bookmarks:
         logger.warning("No bookmarks to export")
-        return "⚠️ No bookmarks to export."
+        return None  # Return None instead of a message
 
     try:
         logger.info("Exporting bookmarks to HTML")
@@ -575,21 +575,24 @@ def export_bookmarks():
             dl.append(dt)
         soup.append(dl)
         html_content = str(soup)
-        b64 = base64.b64encode(html_content.encode()).decode()
-        href = f'data:text/html;base64,{b64}'
+        # Save to a temporary file
+        output_file = "exported_bookmarks.html"
+        with open(output_file, 'w', encoding='utf-8') as f:
+            f.write(html_content)
         logger.info("Bookmarks exported successfully")
-        return f'<a href="{href}" download="bookmarks.html">💾 Download Exported Bookmarks</a>'
+        return output_file  # Return the file path
     except Exception as e:
         logger.error(f"Error exporting bookmarks: {e}", exc_info=True)
-        return "⚠️ Error exporting bookmarks."
+        return None  # Return None in case of error
 
-def chatbot_response(user_query):
+def chatbot_response(user_query, chat_history):
     """
-    Generate chatbot response using the FAISS index and embeddings.
+    Generate chatbot response using the FAISS index and embeddings, maintaining chat history.
     """
     if not bookmarks or faiss_index is None:
         logger.warning("No bookmarks available for chatbot")
-        return "⚠️ No bookmarks available. Please upload and process your bookmarks first."
+        chat_history.append((user_query, "⚠️ No bookmarks available. Please upload and process your bookmarks first."))
+        return chat_history
 
     logger.info(f"Chatbot received query: {user_query}")
 
@@ -607,7 +610,9 @@ def chatbot_response(user_query):
         matching_bookmarks = [id_to_bookmark.get(id) for id in ids if id in id_to_bookmark]
 
         if not matching_bookmarks:
-            return "No relevant bookmarks found for your query."
+            answer = "No relevant bookmarks found for your query."
+            chat_history.append((user_query, answer))
+            return chat_history
 
         # Format the response
         bookmarks_info = "\n".join([
@@ -649,17 +654,21 @@ Provide a concise and helpful response.
         answer = response['choices'][0]['message']['content'].strip()
         logger.info("Chatbot response generated")
         time.sleep(sleep_time)
-        return answer
+
+        # Append the interaction to chat history
+        chat_history.append((user_query, answer))
+        return chat_history
 
     except openai.error.RateLimitError as e:
         wait_time = int(e.headers.get("Retry-After", 5))
         logger.warning(f"Rate limit reached. Waiting for {wait_time} seconds before retrying...")
         time.sleep(wait_time)
-        return chatbot_response(user_query)  # Retry after waiting
+        return chatbot_response(user_query, chat_history)  # Retry after waiting
     except Exception as e:
         error_message = f"⚠️ Error processing your query: {str(e)}"
         logger.error(error_message, exc_info=True)
-        return error_message
+        chat_history.append((user_query, error_message))
+        return chat_history
 
 def build_app():
     """
@@ -702,12 +711,12 @@ def build_app():
                 Ask questions about your bookmarks and get relevant results.
                 """)
 
+                chatbot = gr.Chatbot(label="💬 Chat with SmartMarks")
                 user_input = gr.Textbox(
                     label="✍️ Ask about your bookmarks",
                     placeholder="e.g., Do I have any bookmarks about AI?"
                 )
                 chat_button = gr.Button("📨 Send")
-                chat_output = gr.Textbox(label="💬 Response", interactive=False)
 
             # Manage Bookmarks Tab
             with gr.Tab("Manage Bookmarks"):
@@ -733,7 +742,7 @@ def build_app():
                     edit_category_button = gr.Button("✏️ Edit Category")
                     export_button = gr.Button("💾 Export")
 
-                download_link = gr.HTML(label="📥 Download")
+                download_link = gr.File(label="📥 Download Exported Bookmarks")
 
             # Set up event handlers
             process_button.click(
@@ -744,8 +753,8 @@ def build_app():
 
             chat_button.click(
                 chatbot_response,
-                inputs=user_input,
-                outputs=chat_output
+                inputs=[user_input, chatbot],
+                outputs=chatbot
             )
 
             delete_button.click(