Add new cache delete buttons

main.py

@@ -1,5 +1,5 @@
 
-from os import getenv
+from os import getenv, remove
 from os.path import exists
 from functools import cache
 import json
@@ -52,6 +52,23 @@ def search_results( query ):
     
     return results
 
+def content_summary( url_id, content ):
+    file_path = 'summaries/' + url_id + '.json'
+    if exists( file_path ):
+        with open( file_path, 'r' ) as file:
+            summary = json.load( file )
+    else:
+        try:
+            summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+            summary = summarizer(content, max_length=130, min_length=30, do_sample=False, truncation=True)
+        except Exception as exception:
+            raise exception
+
+        with open( file_path, 'w' ) as file:
+            json.dump( summary, file )
+    
+    return summary
+
 def main():
     st.title('Google Search')
     query = st.text_input('Search query')
@@ -66,39 +83,47 @@ def main():
         number_of_results = len( results )
         st.success( 'Found {} results.'.format( number_of_results ) )
 
+        # if st.button('Search results JSON'):
+        with st.expander("Search results JSON"):
+            st.json( results )
+
         progress_bar = st.progress(0)
 
         # for result in results:
         for index, result in enumerate(results):
-            url_id = uuid.uuid5( uuid.NAMESPACE_URL, result['link'] ).hex
-            st.write(result['link'])
-            st.write(url_id)
+            with st.container():
+                url_id = uuid.uuid5( uuid.NAMESPACE_URL, result['link'] ).hex
+                st.write(result['link'])
+                st.write(url_id)
+                # if st.button('URL HTML'):
+                #     st.json( results )
+                
+                # if st.button('Page content'):
+                #     st.json( results )
 
-            try:
-                content = get_url_content( result['link'] )
-            except Exception as exception:
-                st.exception(exception)
-                progress_bar.progress( ( index + 1 ) / number_of_results )
-                continue
-
-            file_path = 'summaries/' + url_id + '.json'
-            if exists( file_path ):
-                with open( file_path, 'r' ) as file:
-                    summary = json.load( file )
-            else:
                 try:
-                    summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
-                    summary = summarizer(content, max_length=130, min_length=30, do_sample=False, truncation=True)
+                    content = get_url_content( result['link'] )
                 except Exception as exception:
-                    raise exception
+                    st.exception(exception)
+                    progress_bar.progress( ( index + 1 ) / number_of_results )
+                    continue
 
-                with open( file_path, 'w' ) as file:
-                    json.dump( summary, file )
-    
-            for sentence in summary:
-                st.write(sentence['summary_text'])
+                summary = content_summary( url_id, content )
+        
+                for sentence in summary:
+                    st.write(sentence['summary_text'])
+
+                progress_bar.progress( ( index + 1 ) / number_of_results )
+
+                col1, col2 = st.columns([.5,1])
+                with col1:
+                    if st.button('Delete summary cache', key=url_id + 'summary'):
+                        remove( 'summaries/' + url_id + '.json' )
+
+                with col2:
+                    if st.button('Delete content cache', key=url_id + 'content'):
+                        remove( 'page-content/' + url_id + '.txt' )
  
-            progress_bar.progress( ( index + 1 ) / number_of_results )
 
 if __name__ == '__main__':
     main()