Update app.py

app.py

@@ -232,8 +232,8 @@ def main():
 
     # Process 'q' query parameter from the URL
     try:
-        query_params = st.query_params
-        query_list = (query_params.get('q') or query_params.get('query') or [''])
+        query_params = st.experimental_get_query_params()
+        query_list = query_params.get('q') or query_params.get('query') or []
         if query_list:
             search_query = query_list[0]
             if len(search_query) > 1:
@@ -249,14 +249,17 @@ def main():
                 with open(filename, 'w', encoding='utf-8') as f:
                     f.write(markdown_text)
                 st.write(f"Generated file **{filename}** with AI lookup results.")
+                # Clear the query parameters
+                st.experimental_set_query_params()
                 # Update file management sidebar
                 st.experimental_rerun()
     except Exception as e:
         st.write(f"An error occurred while processing query parameters: {e}")
 
     # Handle 'action' and 'query' parameters
-    if 'action' in st.query_params:
-        action_list = st.query_params['action']
+    params = st.experimental_get_query_params()
+    if 'action' in params:
+        action_list = params['action']
         if action_list:
             action = action_list[0]
             if action == 'show_message':
@@ -265,8 +268,8 @@ def main():
                 # Clear query parameters
                 st.experimental_set_query_params()
                 st.experimental_rerun()
-    if 'query' in st.query_params:
-        query_list = st.query_params['query']
+    if 'query' in params:
+        query_list = params['query']
         if query_list:
             query = query_list[0]
             # Display content or image based on the query
@@ -277,24 +280,29 @@ def main():
     file_management_sidebar()
 
     # Display the selected file
-    if 'selected_file' in st.session_state:
-        selected_file = st.session_state.selected_file
+    selected_file = st.session_state.get('selected_file')
+    if selected_file:
         view_mode = st.session_state.get('view_mode', 'view')
-        if view_mode == 'view':
-            st.markdown(f"### Viewing {selected_file}")
-            with open(selected_file, 'r', encoding='utf-8') as f:
-                file_content = f.read()
-            st.markdown(file_content)
-        elif view_mode == 'edit':
-            st.markdown(f"### Editing {selected_file}")
-            with open(selected_file, 'r', encoding='utf-8') as f:
-                file_content = f.read()
-            edited_content = st.text_area("Edit the markdown content", file_content, height=400)
-            if st.button("Save Changes"):
-                with open(selected_file, 'w', encoding='utf-8') as f:
-                    f.write(edited_content)
-                st.success(f"Changes saved to {selected_file}")
-                st.experimental_rerun()
+        if os.path.exists(selected_file):
+            if view_mode == 'view':
+                st.markdown(f"### Viewing {selected_file}")
+                with open(selected_file, 'r', encoding='utf-8') as f:
+                    file_content = f.read()
+                st.markdown(file_content)
+            elif view_mode == 'edit':
+                st.markdown(f"### Editing {selected_file}")
+                with open(selected_file, 'r', encoding='utf-8') as f:
+                    file_content = f.read()
+                edited_content = st.text_area("Edit the markdown content", file_content, height=400)
+                if st.button("Save Changes"):
+                    with open(selected_file, 'w', encoding='utf-8') as f:
+                        f.write(edited_content)
+                    st.success(f"Changes saved to {selected_file}")
+                    st.experimental_rerun()
+        else:
+            st.error("Selected file does not exist.")
+    else:
+        st.info("No file selected.")
 
 if __name__ == "__main__":
     main()