Update app.py

app.py

@@ -1,3 +1,5 @@
+!apt-get install -y chromium-browser chromium-chromedriver libnss3 libxss1 libatk-bridge2.0-0 libgtk-3-0 libgbm-dev
+
 import nest_asyncio
 nest_asyncio.apply()
 
@@ -31,44 +33,39 @@ VERSION_NUMBER = 2
 # Asynchronous function to generate handwriting image via Pyppeteer
 # ----------------------------
 async def _generate_handwriting_image(text_prompt, screenshot_path):
-    """
-    Launches a headless browser, goes to Calligraphr, types the text, 
-    and takes a screenshot of the rendered handwriting.
-    """
-    # Launch Chromium with additional flags for containerized environments
-    browser = await launch(
-        headless=True,
-        handleSIGINT=False,
-        handleSIGTERM=False,
-        handleSIGHUP=False,
-        args=[
-            '--no-sandbox',
-            '--disable-setuid-sandbox',
-            '--disable-dev-shm-usage',
-            '--disable-gpu',
-            '--single-process',
-            '--no-zygote',
-            '--window-size=1920,1080'
-        ]
-    )
     try:
+        browser = await launch(
+            headless=True,
+            executablePath="/usr/bin/chromium-browser",
+            args=[
+                '--no-sandbox',
+                '--disable-setuid-sandbox',
+                '--disable-dev-shm-usage',
+                '--disable-gpu',
+                '--single-process',
+                '--no-zygote',
+                '--window-size=1920,1080'
+            ]
+        )
         page = await browser.newPage()
-        await page.goto('https://www.calligraphr.com/en/font/', {'waitUntil': 'networkidle2'})
-        await page.waitForSelector('#text-input')
+        await page.goto('https://www.calligraphr.com/en/font/', {
+            'waitUntil': 'networkidle2',
+            'timeout': 60000
+        })
+        await page.waitForSelector('#text-input', {'timeout': 30000})
         await page.type('#text-input', text_prompt)
-
-        # Give the page time to render the handwriting
-        await asyncio.sleep(3)
-
-        # Screenshot a portion of the page that should contain the handwriting
+        await asyncio.sleep(5)
         await page.screenshot({
             'path': screenshot_path,
             'clip': {'x': 100, 'y': 200, 'width': 600, 'height': 150}
         })
         return screenshot_path
-
+    except Exception as e:
+        logging.error(f"Pyppeteer error: {str(e)}")
+        return None
     finally:
-        await browser.close()
+        if 'browser' in locals():
+            await browser.close()
 
 def generate_handwriting_image(text_prompt, screenshot_path="/tmp/handwriting.png"):
     """
@@ -218,6 +215,11 @@ interface = gr.Interface(
     allow_flagging="never"
 )
 
+# ... [rest of your existing code] ...
+
 if __name__ == "__main__":
-    logging.debug("Launching Gradio interface.")
-    interface.launch(share=True)
+    interface.launch(
+        server_name="0.0.0.0",
+        server_port=int(os.environ.get("PORT", 7860)),
+        enable_queue=True
+    )