Update app.py

app.py

@@ -213,12 +213,6 @@ def file_management_sidebar():
             st.session_state.selected_file = new_filename
             st.session_state.view_mode = 'edit'
 
-def get_all_query_params(key):
-    return st.query_params().get(key, [])
-
-def clear_query_params():
-    st.query_params()  
-
     
 # Main application logic
 def main():
@@ -239,28 +233,20 @@ 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 []
-        if query_list:
-            search_query = query_list[0]
-            if len(search_query) > 1:
-
-                
-                st.write(f"### Search query received: {search_query}")
-                ai_result = perform_ai_lookup(search_query)
-                markdown_text = extract_urls(ai_result)
-                st.markdown("## Extracted URLs")
-                st.markdown(markdown_text)
-
-                
-                filename = generate_filename("AI_Result", search_query)
-                with open(filename, 'w', encoding='utf-8') as f:
-                    f.write(markdown_text)
-                st.write(f"Generated file **{filename}** with AI lookup results.")
-
-                
-                st.query_params.clear()
-                st.session_state.selected_file = filename
-                st.session_state.view_mode = 'view'
+        query = (query_params.get('q') or query_params.get('query') or [''])
+        if len(query) > 1: 
+            st.write(f"### Search query received: {query}")
+            ai_result = perform_ai_lookup(query)
+            markdown_text = extract_urls(ai_result)
+            st.markdown("## Extracted URLs")
+            st.markdown(markdown_text)
+            filename = generate_filename(ai_result, query)
+            with open(filename, 'w', encoding='utf-8') as f:
+                f.write(markdown_text)
+            st.write(f"Generated file **{filename}** with AI results for {query}")
+            st.query_params.clear()
+            st.session_state.selected_file = filename
+            st.session_state.view_mode = 'view'
                 
     except Exception as e:
         st.write(f"An error occurred while processing query parameters: {e}")
@@ -268,20 +254,19 @@ def main():
     # Handle 'action' and 'query' parameters
     params = st.query_params
     if 'action' in params:
-        action_list = params['action']
-        if action_list:
-            action = action_list[0]
-            if action == 'show_message':
-                st.success("Showing a message because 'action=show_message' was found in the URL.")
-            elif action == 'clear':
-                # Clear query parameters
-                st.query_params.clear()
+        action = st.query_params()['action'][0]
+        if action == 'show_message':
+            st.success("Showing a message because 'action=show_message' was found in the URL.")
+        elif action == 'clear':
+            # Clear query parameters
+            st.query_params.clear()
                 
-    if 'query' in params:
-        query_list = params['query']
-        if query_list:
-            query = query_list[0]
-            st.write(f"Displaying content for query: {query}")
+    if 'query' in st.query_params:
+        query = st.query_params['query'][0]  # Get the query parameter
+        # Display content or image based on the query
+        #display_content_or_image(query)
+        st.markdown("To Implement:")
+        st.markdown(query)
 
     # File management sidebar
     file_management_sidebar()