Update app.py

app.py

@@ -1,38 +1,40 @@
 import os
+import streamlit as st
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from ascii_magic import AsciiArt
 
 def capture_screenshot(url, output_path='screenshot.png'):
-    # Set up headless Chrome
     chrome_options = Options()
     chrome_options.add_argument('--headless')
     chrome_options.add_argument('--disable-gpu')
+    chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--window-size=1920,1080')
 
-    # Initialize WebDriver
     driver = webdriver.Chrome(options=chrome_options)
     driver.get(url)
-
-    # Save screenshot
     driver.save_screenshot(output_path)
     driver.quit()
 
 def convert_to_ascii(image_path):
-    # Create an AsciiArt instance from the image file
-    my_art = AsciiArt.from_image(image_path)
-    # Display the ASCII art in the terminal
-    my_art.to_terminal(columns=100, monochrome=True)
+    ascii_art = AsciiArt.from_image(image_path)
+    return ascii_art.to_string(columns=100, monochrome=True)
+
+st.title("ASCII Web Screenshot Viewer")
+
+url = st.text_input("Enter a website URL:", "https://example.com")
 
-if __name__ == '__main__':
-    url = input("Enter the website URL (e.g., https://example.com): ").strip()
-    screenshot_file = 'screenshot.png'
+if st.button("Generate ASCII Screenshot"):
+    try:
+        screenshot_file = "screenshot.png"
+        st.write("Capturing website screenshot...")
+        capture_screenshot(url, screenshot_file)
 
-    print(f"\nCapturing screenshot of {url}...")
-    capture_screenshot(url, screenshot_file)
+        st.write("Converting to ASCII art...")
+        ascii_output = convert_to_ascii(screenshot_file)
 
-    print("\nConverting screenshot to ASCII art:\n")
-    convert_to_ascii(screenshot_file)
+        st.text_area("ASCII Art Output", ascii_output, height=600)
+        os.remove(screenshot_file)
 
-    # Optional: Remove the screenshot file after conversion
-    os.remove(screenshot_file)
+    except Exception as e:
+        st.error(f"An error occurred: {e}")