Update app2.py

app2.py

@@ -74,11 +74,15 @@ current_state = 'main_panel'  # Variable to keep track of the current state
 # Function to reset the main panel content to the initial content and store the previous content
 def reset_main_panel(event):
     global previous_content, current_state
+    
     # Store the previous content of the current state if it's not already stored
-    if current_state in previous_content:
+    if current_state not in previous_content:
+        current_state = 'main_panel'
+        previous_content[current_state] = main_body_instance.main_body.copy()
+    else:
         previous_content[current_state] = main_body_instance.main_body.copy()
 
-    current_state = 'main_panel'  # Update the current state
+    current_state = 'main_panel'
     # Reset the main panel to the initial state
     main_body_instance.main_body.clear()  # Clear the existing main panel content
     main_body_instance.main_body.append(pn.pane.LaTeX("Please select some tags!", styles={'font-size': '20pt'}))
@@ -88,11 +92,16 @@ def reset_main_panel(event):
 def update_main_panel_button3(event):
     global previous_content, current_state
     # Store the previous content of the current state if it's not already stored
-    if current_state in previous_content:
+    if current_state not in previous_content:
+        current_state = 'main_panel_button3'
         previous_content[current_state] = main_body_instance.main_body.copy()
+    else:
+        previous_content[current_state] = main_body_instance.main_body.copy()
+
+    # Update the current state to main_panel_button3
+    current_state = 'main_panel_button3'
 
-    current_state = 'main_panel_button3'  # Update the current state
-    # Update the main panel to the desired content when button3 is pressed
+    # Reset the main panel to the desired content when button3 is pressed
     main_body_instance.main_body.clear()  # Clear the existing main panel content
     main_body_instance.main_body.append(pn.pane.Markdown("You pressed button3!"))
     main_body_instance.param.trigger('main_body')  # Trigger the update manually
@@ -105,17 +114,13 @@ button3.on_click(update_main_panel_button3)
 def show_previous_content(event):
     global previous_content, current_state
     # Check if there's previous content for the current state
-    print(current_state, flush=True)
+    print(current_state, flush = True)
     if current_state in previous_content:
         main_body_instance.main_body.clear()  # Clear the existing main panel content
         main_body_instance.main_body.extend(previous_content[current_state])  # Add the previous content
     else:
         main_body_instance.main_body.clear()  # Clear the existing main panel content
         main_body_instance.main_body.append(pn.pane.Markdown("No previous content available!"))
-
-    # Update the current state before displaying the previous content
-    current_state = 'main_panel' if current_state == 'main_panel_button3' else current_state
-
     main_body_instance.param.trigger('main_body')  # Trigger the update manually
 
 # Attach the show_previous_content function to the on_click event of button1 and button3