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
-import ascii_magic
+from ascii_magic import AsciiArt
 
 # Function to capture a screenshot of a webpage
 def capture_screenshot(url, output_path='screenshot.png'):
@@ -23,7 +23,7 @@ def capture_screenshot(url, output_path='screenshot.png'):
 def convert_to_ascii(image_path):
     try:
         # Correct usage of ascii_magic to convert image to ASCII
-        ascii_art = ascii_magic.to_ascii_image(image_path, columns=100, mode='mono')
+        ascii_art = AsciiArt.from_image(image_path, columns=100, mode='mono')
         return ascii_art
     except Exception as e:
         raise Exception(f"Failed to convert image to ASCII: {e}")
@@ -42,7 +42,7 @@ if st.button("Generate ASCII Screenshot"):
         capture_screenshot(url, screenshot_file)
 
         st.write("Converting to ASCII art...")
-        ascii_output = convert_to_ascii(screenshot_file)
+        ascii_output = AsciiArt.from_image(screenshot_file)
 
         # Display ASCII art in a text area
         st.text_area("ASCII Art Output", ascii_output, height=600)