Update app.py

app.py

@@ -469,23 +469,23 @@ def process_uploaded_file(file, state_bookmarks):
 
     if file is None:
         logger.warning("No file uploaded")
-        return "Please upload a bookmarks HTML file.", '', state_bookmarks, display_bookmarks()
+        return "Please upload a bookmarks HTML file.", '', state_bookmarks, display_bookmarks(), gr.update(choices=[])
 
     try:
         file_content = file.decode('utf-8')
     except UnicodeDecodeError as e:
         logger.error(f"Error decoding the file: {e}", exc_info=True)
-        return "Error decoding the file. Please ensure it's a valid HTML file.", '', state_bookmarks, display_bookmarks()
+        return "Error decoding the file. Please ensure it's a valid HTML file.", '', state_bookmarks, display_bookmarks(), gr.update(choices=[])
 
     try:
         bookmarks = parse_bookmarks(file_content)
     except Exception as e:
         logger.error(f"Error parsing bookmarks: {e}", exc_info=True)
-        return "Error parsing the bookmarks HTML file.", '', state_bookmarks, display_bookmarks()
+        return "Error parsing the bookmarks HTML file.", '', state_bookmarks, display_bookmarks(), gr.update(choices=[])
 
     if not bookmarks:
         logger.warning("No bookmarks found in the uploaded file")
-        return "No bookmarks found in the uploaded file.", '', state_bookmarks, display_bookmarks()
+        return "No bookmarks found in the uploaded file.", '', state_bookmarks, display_bookmarks(), gr.update(choices=[])
 
     # Assign unique IDs to bookmarks
     for idx, bookmark in enumerate(bookmarks):
@@ -505,7 +505,7 @@ def process_uploaded_file(file, state_bookmarks):
         faiss_index = vectorize_and_index(bookmarks)
     except Exception as e:
         logger.error(f"Error building FAISS index: {e}", exc_info=True)
-        return "Error building search index.", '', state_bookmarks, display_bookmarks()
+        return "Error building search index.", '', state_bookmarks, display_bookmarks(), gr.update(choices=[])
 
     message = f"✅ Successfully processed {len(bookmarks)} bookmarks."
     logger.info(message)
@@ -518,7 +518,7 @@ def process_uploaded_file(file, state_bookmarks):
     # Update state
     state_bookmarks = bookmarks.copy()
 
-    return message, bookmark_html, state_bookmarks, bookmark_html
+    return message, bookmark_html, state_bookmarks, bookmark_html, gr.update(choices=choices)
 
 def delete_selected_bookmarks(selected_indices, state_bookmarks):
     """
@@ -625,6 +625,9 @@ def chatbot_response(user_query, chat_history):
     logger.info(f"Chatbot received query: {user_query}")
 
     try:
+        # Append user's message to chat history
+        chat_history.append({"role": "user", "content": user_query})
+
         # Rate Limiting Logic
         with api_lock:
             global last_api_call_time
@@ -694,7 +697,7 @@ Provide a concise and helpful response.
         logger.info("Chatbot response generated")
         time.sleep(sleep_time)
 
-        # Append the interaction to chat history
+        # Append the assistant's response to chat history
         chat_history.append({"role": "assistant", "content": answer})
         return chat_history
 
@@ -765,7 +768,7 @@ Navigate through the tabs to explore each feature in detail.
                 process_button.click(
                     process_uploaded_file,
                     inputs=[upload, state_bookmarks],
-                    outputs=[output_text, bookmark_display, state_bookmarks, bookmark_display]
+                    outputs=[output_text, bookmark_display, state_bookmarks, bookmark_display, gr.update()]
                 )
 
             # Chat with Bookmarks Tab