Rename project to "capture-page-tools" and implement Gradio app for webpage screenshot capture

Author: npv2k1

.github/workflows/deploy-to-huggingface.yml

@@ -43,7 +43,7 @@ jobs:
           SPACE_NAME: ${{ secrets.SPACE_NAME }}
         run: |
           # Create repository URL
-          REPO_URL="https://huggingface.co/spaces/$HF_USERNAME/template-python"
+          REPO_URL="https://huggingface.co/spaces/$HF_USERNAME/capture-page-tools"
           
           # Add Hugging Face as a remote and push
           git remote add space $REPO_URL || git remote set-url space $REPO_URL

Dockerfile

@@ -32,7 +32,7 @@ ENTRYPOINT []
 
 RUN python setup.py
 
-EXPOSE 8080
+EXPOSE 7860
 # Run the FastAPI application by default
 # Uses `fastapi dev` to enable hot-reloading when the `watch` sync occurs
 # Uses `--host 0.0.0.0` to allow access from outside the container

main.py

@@ -10,7 +10,7 @@ def main() -> None:
     """Main application entry point."""
     try:
         print("Start app...")
-        import src.modules.api
+        import src.modules.apps
 
     except Exception as e:
         logging.error(f"Application failed to start: {e}", exc_info=True)

pyproject.toml

@@ -1,14 +1,16 @@
 [project]
-name = "template-python"
+name = "capture-page-tools"
 version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "fastapi[standard]>=0.115.8",
+    "gradio>=5.3.0",
     "ipykernel>=6.29.5",
     "numpy>=2.2.3",
     "pytest>=8.3.4",
+    "selenium>=4.29.0",
 ]
 
 [tool.black]

src/modules/apps/__init__.py

@@ -0,0 +1,89 @@
+import gradio as gr
+import tempfile
+import os
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+print("Gradio app loaded.")
+def capture_page(url: str, output_file: str = "screenshot.png"):
+    """
+    Captures a screenshot of the given webpage.
+    
+    :param url: The URL of the webpage to capture.
+    :param output_file: The filename to save the screenshot.
+    """
+    options = Options()
+    options.add_argument("--headless")  # Run in headless mode
+    options.add_argument("--window-size=1920,1080")  # Set window size
+
+    driver = webdriver.Chrome(options=options)
+    
+    try:
+        driver.get(url)
+        driver.save_screenshot(output_file)
+        print(f"Screenshot saved: {output_file}")
+    finally:
+        driver.quit()
+
+def capture_and_show(url: str):
+    """Capture webpage and return the image"""
+    try:
+        # Create temporary file for the screenshot
+        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp:
+            temp_path = tmp.name
+        
+        # Capture the webpage
+        capture_page(url, temp_path)
+        
+        # Return the image path
+        return temp_path
+    except Exception as e:
+        return f"Error capturing page: {str(e)}"
+    
+def create_gradio_app():
+    """Create the main Gradio application with all components"""
+    with gr.Blocks() as app:
+        gr.Markdown("# Webpage Screenshot Capture")
+        
+        with gr.Row():
+            url_input = gr.Textbox(
+                label="Website URL",
+                placeholder="Enter website URL (e.g., https://www.example.com)",
+                scale=4
+            )
+            capture_btn = gr.Button("Capture", scale=1)
+            
+        with gr.Row():
+            output_image = gr.Image(
+                label="Captured Screenshot",
+                type="filepath"
+            )
+            
+        # Connect the components
+        capture_btn.click(
+            fn=capture_and_show,
+            inputs=[url_input],
+            outputs=[output_image]
+        )
+        
+    return app
+
+app = create_gradio_app()
+
+# Configure server settings for Docker deployment
+server_port = 7860  # Standard Gradio port
+server_name = "0.0.0.0"  # Allow external connections
+
+def main():
+    """Launch the Gradio application"""
+    print("Starting Gradio server...")
+    app.launch(
+        server_name=server_name,
+        server_port=server_port,
+        share=False,  # Disable sharing as we're running in Docker
+        auth=None,    # Can be configured if authentication is needed
+        ssl_verify=False,  # Disable SSL verification for internal Docker network
+        show_error=True,
+        favicon_path=None
+    )
+
+main()

src/utils/__init__.py

@@ -1,4 +1,4 @@
 """
 Main application package.
 This is the root package of the application.
-"""
+"""