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
+import ascii_magic
 
 def capture_screenshot(url, output_path='screenshot.png'):
     chrome_options = Options()
@@ -17,8 +17,8 @@ def capture_screenshot(url, output_path='screenshot.png'):
     driver.quit()
 
 def convert_to_ascii(image_path):
-    ascii_art = AsciiArt.from_image(image_path)
-    return ascii_art.to_string(columns=100, monochrome=True)
+    output = ascii_magic.from_image_path(image_path, columns=100, mode=ascii_magic.Modes.ASCII)
+    return str(output)
 
 st.title("ASCII Web Screenshot Viewer")