Update app.py

app.py

@@ -1,23 +1,53 @@
+import asyncio
 import streamlit as st
-from requests_html import HTMLSession
+from requests_html import AsyncHTMLSession
 
-def perform_search(search_query):
-    session = HTMLSession()
-    url = f'https://www.bing.com/search?q={search_query}'
-    response = session.get(url)
-    response.html.render(sleep=1)
-    search_results = response.html.find('li.b_algo')
+async def perform_search(search_query, search_engine):
+    session = AsyncHTMLSession()
+    if search_engine == "Google":
+        url = f"https://www.google.com/search?q={search_query}"
+    elif search_engine == "Bing":
+        url = f"https://www.bing.com/search?q={search_query}"
+    response = await session.get(url)
+    await response.html.arender(sleep=1)
+    
+    # Extract search results based on the search engine
+    if search_engine == "Google":
+        search_results = response.html.find(".yuRUbf a")
+    elif search_engine == "Bing":
+        search_results = response.html.find(".b_algo h2 a")
+    
     results = []
     for result in search_results:
-        title = result.find('h2', first=True).text
-        link = result.find('a', first=True).attrs['href']
-        results.append(f'{title}\n{link}')
+        title = result.text
+        link = result.attrs["href"]
+        results.append({"title": title, "link": link})
+    
     return results
 
-st.title('Web Automation with requests-html')
-search_query = st.text_input('Enter a search query')
-if st.button('Search'):
-    results = perform_search(search_query)
-    st.write('Search Results:')
-    for result in results:
-        st.write(result)
+def main():
+    st.set_page_config(page_title="Web Search App", page_icon=":mag:", layout="wide")
+    
+    st.title("Web Search App")
+    st.write("Search Google and Bing simultaneously!")
+    
+    search_query = st.text_input("Enter your search query")
+    
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        st.header("Google Search Results")
+        if st.button("Search Google"):
+            google_results = asyncio.run(perform_search(search_query, "Google"))
+            for result in google_results:
+                st.write(f"[{result['title']}]({result['link']})")
+    
+    with col2:
+        st.header("Bing Search Results")
+        if st.button("Search Bing"):
+            bing_results = asyncio.run(perform_search(search_query, "Bing"))
+            for result in bing_results:
+                st.write(f"[{result['title']}]({result['link']})")
+
+if __name__ == "__main__":
+    main()