Update app.py

app.py

@@ -1,42 +1,20 @@
 import streamlit as st
-from extract import take_webdata
-from PIL import Image
-from io import BytesIO
+from playwright.sync_api import sync_playwright
 
-def main():
-    st.title("Website Content Exctractor")
-    
-    # Get website URL from user input
-    url = st.text_input("Enter a URL:", "")
-    if st.button("Proceed"):
-        if not url:
-            st.warning("URL is empty.")
-        else:
-            visualize(url)
-  
+st.title("🌐 Web Browser ងាយស្រួលប្រើ")
 
-def visualize(url):  
-    try:
-    # Fetch and display the website content
-        with st.spinner("loading website data ..."):
-            # innerHTML = get_innerHTML(url)
-            html_image, html_content = take_webdata(url)
-            st.subheader("Website title:")
-            if html_content:
-                st.info(html_content)
-            else:
-                st.error("Error: empty html content")
-            st.subheader("Website preview:")
-            if html_image:
-                st.image(html_image)
-            else:
-                st.error("Error: empty html preview")
-                                   
-    
-    except Exception as e:
-        st.error(f"Error: {e}")
+url = st.text_input("បញ្ចូល Link វេបសាយ:", "https://www.google.com")
 
-
-
-if __name__ == "__main__":
-    main()
+if st.button("ទៅកាន់វេបសាយ"):
+    with sync_playwright() as p:
+        # បើក Browser ងាយៗ
+        browser = p.chromium.launch(headless=True)
+        page = browser.new_page()
+        page.goto(url)
+        
+        # ថតរូបបង្ហាញ
+        st.success(f"កំពុងបង្ហាញវេបសាយ: {url}")
+        screenshot = page.screenshot(full_page=False)
+        st.image(screenshot, caption="រូបភាពអេក្រង់")
+        
+        browser.close()