Update app.py

app.py

@@ -46,10 +46,30 @@ ankban@wharton.upenn.edu""")
             )
 
         # === Button Hooks ===
-        btn_spoken.click(lambda: "spoken", outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode], fn=switch_mode)
-        btn_written.click(lambda: "written", outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode], fn=switch_mode)
-        btn_file_study.click(lambda: "study", outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode], fn=switch_mode)
-        btn_file_view.click(lambda: "file", outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode], fn=switch_mode)
+        btn_spoken.click(
+            fn=switch_mode,
+            inputs=[gr.State("spoken")],
+            outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode]
+        )
+        
+        btn_written.click(
+            fn=switch_mode,
+            inputs=[gr.State("written")],
+            outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode]
+        )
+        
+        btn_file_study.click(
+            fn=switch_mode,
+            inputs=[gr.State("study")],
+            outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode]
+        )
+        
+        btn_file_view.click(
+            fn=switch_mode,
+            inputs=[gr.State("file")],
+            outputs=[spoken_panel, written_panel, study_panel, file_panel, current_mode]
+        )
+
 
     return app