Add progress bar and seach result numbers

main.py

@@ -62,34 +62,43 @@ def main():
             except Exception as exception:
                 st.exception(exception)
                 return
-                
 
-            for result in results:
-                url_id = uuid.uuid5( uuid.NAMESPACE_URL, result['link'] ).hex
-                st.write(result['link'])
-                st.write(url_id)
+        number_of_results = len( results )
+        st.success( 'Found {} results.'.format( number_of_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)
+
+            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:
-                    content = get_url_content( result['link'] )
+                    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:
-                    st.exception(exception)
-                    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)
-                    except Exception as exception:
-                        raise exception
-                    with open( file_path, 'w' ) as file:
-                        json.dump( summary, file )
-        
-                for sentence in summary:
-                    st.write(sentence['summary_text'])
+                    raise exception
+
+                with open( file_path, 'w' ) as file:
+                    json.dump( summary, file )
+    
+            for sentence in summary:
+                st.write(sentence['summary_text'])
+ 
+            progress_bar.progress( ( index + 1 ) / number_of_results )
 
 if __name__ == '__main__':
     main()