Update app.py

app.py

@@ -2,7 +2,7 @@ import requests
 from bs4 import BeautifulSoup
 import streamlit as st
 
-# Custom CSS for clean UI
+# CSS for UI
 def load_css():
     st.markdown("""
     <style>
@@ -54,16 +54,6 @@ def load_css():
         color: #545454;
         font-size: 14px;
     }
-    .iframe-container {
-        margin-top: 20px;
-        text-align: center;
-    }
-    .iframe-container iframe {
-        width: 80%;
-        height: 600px;
-        border: 1px solid #ccc;
-        border-radius: 8px;
-    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -92,7 +82,7 @@ def parse_search_results(html_content):
 
 # Streamlit App
 def main():
-    st.set_page_config(page_title="Google Search Clone with Safe Navigation", layout="centered")
+    st.set_page_config(page_title="Google Search Clone with Secure Navigation", layout="centered")
     load_css()
     
     st.markdown('<div class="search-bar">', unsafe_allow_html=True)
@@ -110,21 +100,28 @@ def main():
             
             if html_content:
                 search_results = parse_search_results(html_content)
-                for idx, result in enumerate(search_results, start=1):
+                for result in search_results:
                     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>
+                        <a href="/safe_preview?url={result['link']}" target="_self">Secure Preview</a>
                     </div>
                     """, unsafe_allow_html=True)
 
-    # Iframe secure preview for websites
-    if "safe_view" in st.query_params:
+    # Display website securely
+    if "safe_preview" in st.query_params:
         url = st.query_params.get("url", [None])[0]
         if url:
-            st.markdown(f"<div class='iframe-container'><iframe src='{url}'></iframe></div>", unsafe_allow_html=True)
+            try:
+                response = requests.get(url, timeout=10)
+                if response.status_code == 200:
+                    st.write(f"### Preview of: {url}")
+                    st.code(response.text[:2000], language='html')  # Display HTML preview
+                else:
+                    st.error(f"Failed to fetch the website. Status code: {response.status_code}")
+            except Exception as e:
+                st.error(f"Error fetching the website: {str(e)}")
 
 if __name__ == "__main__":
     main()