Update app.py

app.py

@@ -1,19 +1,26 @@
-import streamlit as st 
-from googlesearch import search 
-import requests 
-from bs4 import BeautifulSoup 
+import streamlit as st
+from googlesearch import search
+import requests
+from bs4 import BeautifulSoup
 
-# Function to perform Google search and return the first link
+# Function to perform Google search and return the first two links
 def google_search(query):
     try:
         query = query + "/t site:https://medium.com/"
         # Perform the search and get an iterator of results
         search_results = search(query, num_results=10)  # Get up to 10 results
-        first_link = next(search_results, None)  # Get the first result
-        return first_link
+        first_two_links = []
+        
+        # Get the first two results
+        for i, link in enumerate(search_results):
+            if i < 2:
+                first_two_links.append(link)
+            else:
+                break
+        return first_two_links
     except Exception as e:
         st.error(f"An error occurred: {e}")
-        return None  
+        return None
 
 # Function to fetch webpage content
 def fetch_webpage_content(url):
@@ -23,7 +30,7 @@ def fetch_webpage_content(url):
         return response.text
     except Exception as e:
         st.error(f"Failed to fetch the webpage content: {e}")
-        return None  
+        return None
 
 # Function to scrape text from webpage content using Beautiful Soup
 def scrape_text(webpage_content):
@@ -43,35 +50,38 @@ def scrape_text(webpage_content):
         return text
     except Exception as e:
         st.error(f"Failed to scrape text from webpage content: {e}")
-        return None  
+        return None
 
 # Streamlit app UI
-st.title("Search Link Finder")  
+st.title("Search Link Finder")
 
 # Input field for search query
-query = st.text_input("Enter search query", "")  
+query = st.text_input("Enter search query", "")
 
 # Button to trigger search
 if st.button("Search"):
     if query:
-        first_link = google_search(query)
-        if first_link:
-            st.success(f"First link: [Click here]({first_link})")  
-            # Fetch webpage content
-            webpage_content = fetch_webpage_content(first_link)  
-            if webpage_content:
-                # Scrape text from webpage content
-                scraped_text = scrape_text(webpage_content)  
-                if scraped_text:
-                    st.write(scraped_text)  
-                # Download button for the webpage content
-                st.download_button(
-                    label="Download Webpage Content",
-                    data=scraped_text,
-                    file_name="webpage_content.txt",
-                    mime="text/plain"
-                )  
+        first_two_links = google_search(query)
+        if first_two_links:
+            for i, link in enumerate(first_two_links):
+                st.success(f"Link {i+1}: [Click here]({link})")
+                
+                # Fetch webpage content
+                webpage_content = fetch_webpage_content(link)
+                if webpage_content:
+                    # Scrape text from webpage content
+                    scraped_text = scrape_text(webpage_content)
+                    if scraped_text:
+                        st.write(f"Scraped Content from Link {i+1}:")
+                        st.write(scraped_text)
+                        # Download button for the webpage content
+                        st.download_button(
+                            label=f"Download Webpage Content from Link {i+1}",
+                            data=scraped_text,
+                            file_name=f"webpage_content_{i+1}.txt",
+                            mime="text/plain"
+                        )
         else:
-            st.warning("No results found")  
+            st.warning("No results found")
     else:
-        st.error("Please enter a query")
+        st.error("Please enter a query")