Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import streamlit as st
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
-from ascii_magic import AsciiArt
+from PIL import Image
 
 # Function to capture a screenshot of a webpage
 def capture_screenshot(url, output_path='screenshot.png'):
@@ -19,33 +19,21 @@ def capture_screenshot(url, output_path='screenshot.png'):
     finally:
         driver.quit()  # Ensure the driver is closed even if an error occurs
 
-# Function to convert an image to ASCII art
-def convert_to_ascii(image_path):
-    try:
-        # Create ASCII art from the image
-        ascii_art = AsciiArt.from_image(image_path)  # Create the AsciiArt object
-        return ascii_art.get_ascii()  # Use the get_ascii() method to get the ASCII representation
-    except Exception as e:
-        raise Exception(f"Failed to convert image to ASCII: {e}")
-
 # Streamlit app layout
-st.title("ASCII Web Screenshot Viewer")
+st.title("Web Screenshot Viewer")
 
 # Input field for URL
 url = st.text_input("Enter a website URL:", "https://example.com")
 
-# Button to trigger the screenshot and conversion process
-if st.button("Generate ASCII Screenshot"):
+# Button to trigger the screenshot process
+if st.button("Generate Screenshot"):
     try:
         screenshot_file = "screenshot.png"
         st.write("Capturing website screenshot...")
         capture_screenshot(url.strip('"'))  # Strip any extra quotes from the URL
 
-        st.write("Converting to ASCII art...")
-        ascii_output = convert_to_ascii(screenshot_file)  # Use the correct function
-
-        # Display ASCII art in a text area
-        st.text_area("ASCII Art Output", ascii_output, height=600)
+        # Display the screenshot in the Streamlit app
+        st.image(screenshot_file, caption='Screenshot of the webpage', use_column_width=True)
 
     except Exception as e:
         st.error(f"An error occurred: {e}")