Update app.py

app.py

@@ -1,6 +1,6 @@
 # ==============================================
 # SCREENSHOT SCRAPER FOR N8N
-# Complete version with all improvements - GRADIO 4.x COMPATIBLE
+# Complete version with all improvements - GRADIO COMPATIBLE
 # ==============================================
 
 import gradio as gr
@@ -558,7 +558,8 @@ scraper = ScreenshotScraper()
 # FASTAPI APPLICATION
 # ==============================================
 
-app = FastAPI(
+# Create FastAPI app
+fastapi_app = FastAPI(
     title="Screenshot Scraper API",
     description="AI-powered web scraper that takes screenshots and extracts text using OCR",
     version="2.0.0"
@@ -566,7 +567,7 @@ app = FastAPI(
 
 # CORS middleware
 from fastapi.middleware.cors import CORSMiddleware
-app.add_middleware(
+fastapi_app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
@@ -574,7 +575,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-@app.get("/")
+@fastapi_app.get("/")
 async def root():
     """Root endpoint with API information"""
     return {
@@ -593,7 +594,7 @@ async def root():
         }
     }
 
-@app.get("/health")
+@fastapi_app.get("/health")
 async def health():
     """Health check endpoint"""
     return {
@@ -605,7 +606,7 @@ async def health():
         }
     }
 
-@app.get("/api/info")
+@fastapi_app.get("/api/info")
 async def api_info():
     """Get system information"""
     return {
@@ -619,7 +620,7 @@ async def api_info():
         }
     }
 
-@app.post("/api/scrape")
+@fastapi_app.post("/api/scrape")
 async def api_scrape(data: dict = None, url: str = None):
     """
     Main scraping endpoint for n8n
@@ -676,10 +677,10 @@ async def api_scrape(data: dict = None, url: str = None):
         }
 
 # ==============================================
-# GRADIO INTERFACE (UPDATED FOR GRADIO 4.x)
+# GRADIO INTERFACE
 # ==============================================
 
-def gradio_scrape(url: str, use_html_fallback: bool = False):
+def gradio_scrape(url: str):
     """Gradio interface function"""
     if not url:
         return "❌ Please enter a URL", {"error": "No URL provided"}
@@ -718,22 +719,15 @@ def gradio_scrape(url: str, use_html_fallback: bool = False):
         error_msg = f"## ❌ Unexpected Error\n\n{str(e)}"
         return error_msg, {"error": str(e), "url": url}
 
-# Create Gradio interface with Gradio 4.x compatible settings
-iface = gr.Interface(
+# Create Gradio interface
+gradio_app = gr.Interface(
     fn=gradio_scrape,
-    inputs=[
-        gr.Textbox(
-            label="Website URL",
-            placeholder="https://example.com",
-            value="https://example.com",
-            lines=1
-        ),
-        gr.Checkbox(
-            label="Use HTML fallback (if available)",
-            value=False,
-            visible=False  # Hidden for now
-        )
-    ],
+    inputs=gr.Textbox(
+        label="Website URL",
+        placeholder="https://example.com",
+        value="https://example.com",
+        lines=1
+    ),
     outputs=[
         gr.Markdown(label="Result"),
         gr.JSON(label="API Response")
@@ -753,21 +747,15 @@ iface = gr.Interface(
         ["https://en.wikipedia.org/wiki/Artificial_intelligence"],
         ["https://news.ycombinator.com"],
         ["https://httpbin.org/html"]
-    ],
-    theme="soft"
-    # Removed allow_flagging parameter for Gradio 4.x compatibility
+    ]
 )
 
 # ==============================================
-# GRADIO LAUNCH CONFIGURATION
+# MOUNT GRADIO TO FASTAPI
 # ==============================================
 
-def launch_gradio():
-    """Launch Gradio interface"""
-    # Mount Gradio app to FastAPI
-    app = gr.mount_gradio_app(app, iface, path="/")
-    
-    return app
+# Mount Gradio app to FastAPI
+app = gr.mount_gradio_app(fastapi_app, gradio_app, path="/")
 
 # ==============================================
 # APPLICATION STARTUP
@@ -782,9 +770,6 @@ if __name__ == "__main__":
     print(f"🔧 Default OCR Model: {Config.DEFAULT_MODEL}")
     print("="*60 + "\n")
     
-    # Configure Gradio launch
-    app = launch_gradio()
-    
     uvicorn.run(
         app,
         host="0.0.0.0",