Update app.py

app.py

@@ -1,4 +1,8 @@
-import gradio as gr
+import dash
+from dash import dcc, html, Input, Output, State
+import dash_bootstrap_components as dbc
+from dash.exceptions import PreventUpdate
+import base64
 import requests
 from bs4 import BeautifulSoup
 from urllib.parse import urljoin, urlparse
@@ -13,7 +17,13 @@ import sqlite3
 from contextlib import contextmanager
 from threading import local
 import time
+import os
 
+# Initialize Dash app
+app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
+server = app.server
+
+# Logging setup
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -166,19 +176,68 @@ async def process_url(url, depth):
         logger.error(f"Error in process_url: {str(e)}")
         return f"An error occurred: {str(e)}"
 
-def run_async(url, depth):
-    return asyncio.run(process_url(url, depth))
-
-iface = gr.Interface(
-    fn=run_async,
-    inputs=[
-        gr.Textbox(label="Enter website URL (e.g., https://www.gradio.app/docs)"),
-        gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Crawl Depth")
-    ],
-    outputs=gr.File(label="Download PDF"),
-    title="Website to PDF Converter",
-    description="Enter docs URL and crawl depth to convert documentation pages into a PDF. Be responsible for sites you have permission to do this"
+# App layout
+app.layout = dbc.Container([
+    dbc.Navbar(
+        dbc.Container([
+            html.A(
+                dbc.Row([
+                    dbc.Col(html.Img(src="/assets/logo.png", height="30px")),
+                    dbc.Col(dbc.NavbarBrand("Website to PDF Converter", className="ms-2")),
+                ],
+                align="center",
+                className="g-0",
+                ),
+                href="/",
+                style={"textDecoration": "none"},
+            )
+        ]),
+        color="#116F70",
+        dark=True,
+    ),
+    
+    dbc.Card(
+        dbc.CardBody([
+            html.H1("Website to PDF Converter", className="text-center mb-4"),
+            html.P("Enter docs URL and crawl depth to convert documentation pages into a PDF. Be responsible for sites you have permission to do this", className="text-center mb-4"),
+            dbc.Input(id="url-input", type="text", placeholder="Enter website URL (e.g., https://www.gradio.app/docs)", className="mb-3"),
+            dcc.Slider(id="depth-slider", min=1, max=10, step=1, value=3, marks={i: str(i) for i in range(1, 11)}, className="mb-3"),
+            dbc.Button("Convert to PDF", id="submit-button", color="primary", className="mb-3 w-100"),
+            dbc.Spinner(html.Div(id="output-area"), color="primary", type="grow"),
+        ]),
+        className="mt-4"
+    )
+], fluid=True)
+
+@app.callback(
+    Output("output-area", "children"),
+    Input("submit-button", "n_clicks"),
+    State("url-input", "value"),
+    State("depth-slider", "value"),
+    prevent_initial_call=True
 )
-
-if __name__ == "__main__":
-    iface.launch()
+def update_output(n_clicks, url, depth):
+    if not url:
+        return "Please enter a valid URL."
+    
+    pdf_path = asyncio.run(process_url(url, depth))
+    
+    if pdf_path.startswith("An error occurred"):
+        return pdf_path
+    
+    with open(pdf_path, "rb") as f:
+        encoded = base64.b64encode(f.read()).decode()
+    
+    os.unlink(pdf_path)  # Remove the temporary file
+    
+    return html.Div([
+        html.H4("PDF Generated Successfully"),
+        html.A(
+            dbc.Button("Download PDF", color="success", className="mt-2"),
+            href=f"data:application/pdf;base64,{encoded}",
+            download="website_content.pdf"
+        )
+    ])
+
+if __name__ == '__main__':
+    app.run_server(debug=True, host='0.0.0.0', port=7860)