Update app.py

app.py

@@ -2,7 +2,7 @@ import requests
 from bs4 import BeautifulSoup
 import streamlit as st
 
-# Custom CSS to mimic Google's interface
+# Custom CSS for clean UI
 def load_css():
     st.markdown("""
     <style>
@@ -54,12 +54,15 @@ def load_css():
         color: #545454;
         font-size: 14px;
     }
-    .captcha-warning {
+    .iframe-container {
+        margin-top: 20px;
         text-align: center;
-        margin-top: 50px;
     }
-    .captcha-warning h3 {
-        color: #d93025;
+    .iframe-container iframe {
+        width: 80%;
+        height: 600px;
+        border: 1px solid #ccc;
+        border-radius: 8px;
     }
     </style>
     """, unsafe_allow_html=True)
@@ -72,7 +75,7 @@ def search_google(query, proxy=None):
     url = f'https://www.google.com/search?q={query.replace(" ", "+")}&hl=en'
     proxies = {"http": proxy, "https": proxy} if proxy else None
     response = requests.get(url, headers=headers, proxies=proxies, timeout=10)
-    if "detected unusual traffic" in response.text.lower():  # Detect CAPTCHA
+    if "detected unusual traffic" in response.text.lower():
         return None, True
     return response.text, False
 
@@ -89,7 +92,7 @@ def parse_search_results(html_content):
 
 # Streamlit App
 def main():
-    st.set_page_config(page_title="Google Search Clone", layout="centered")
+    st.set_page_config(page_title="Google Search Clone with Safe Navigation", layout="centered")
     load_css()
     
     st.markdown('<div class="search-bar">', unsafe_allow_html=True)
@@ -102,23 +105,26 @@ def main():
             html_content, captcha_detected = search_google(query)
             
             if captcha_detected:
-                st.markdown("""
-                <div class="captcha-warning">
-                    <h3>CAPTCHA Detected</h3>
-                    <p>Google has detected unusual traffic from this application. Please solve the CAPTCHA manually or try again later.</p>
-                </div>
-                """, unsafe_allow_html=True)
+                st.error("CAPTCHA detected. Please try again later or reduce search frequency.")
                 return
             
             if html_content:
                 search_results = parse_search_results(html_content)
-                for result in search_results:
+                for idx, result in enumerate(search_results, start=1):
                     st.markdown(f"""
                     <div class="search-result">
                         <h3><a href="{result['link']}" target="_blank">{result['title']}</a></h3>
                         <p>{result['snippet']}</p>
+                        <button onclick="window.open('{result['link']}', '_blank')">Visit Website</button>
+                        <button onclick="window.location.href='/safe_view?url={result['link']}'">Preview Securely</button>
                     </div>
                     """, unsafe_allow_html=True)
 
+    # Iframe secure preview for websites
+    if "safe_view" in st.experimental_get_query_params():
+        url = st.experimental_get_query_params().get("url", [None])[0]
+        if url:
+            st.markdown(f"<div class='iframe-container'><iframe src='{url}'></iframe></div>", unsafe_allow_html=True)
+
 if __name__ == "__main__":
     main()