query-app / app.py
Makima57's picture
Upload app.py with huggingface_hub
73a33e7 verified
raw
history blame
1.71 kB
# app.py
import streamlit as st
from googlesearch import search
import requests
# Function to perform Google search and return the first link
def google_search(query):
try:
# 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
except Exception as e:
st.error(f"An error occurred: {e}")
return None
# Function to fetch webpage content
def fetch_webpage_content(url):
try:
response = requests.get(url)
response.raise_for_status() # Check if the request was successful
return response.text
except Exception as e:
st.error(f"Failed to fetch the webpage content: {e}")
return None
# Streamlit app UI
st.title("Search Link Finder")
# Input field for 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:
# Download button for the webpage content
st.download_button(
label="Download Webpage Content",
data=webpage_content,
file_name="webpage_content.html",
mime="text/html"
)
else:
st.warning("No results found")
else:
st.error("Please enter a query")