commmit Dockerfile

Dockerfile

@@ -1,50 +1,64 @@
-FROM python:3.9
+# Use a Python base image
+FROM python:3.9-slim
 
-WORKDIR /code
-
-COPY ./requirements.txt /code/requirements.txt
-
-# Install required system dependencies
+# Install necessary system dependencies
 RUN apt-get update && apt-get install -y \
-    libnss3 \
-    libnspr4 \
-    libdbus-1-3 \
-    libatk1.0-0 \
+    curl \
+    gnupg \
+    ca-certificates \
+    wget \
+    unzip \
+    fonts-liberation \
+    libasound2 \
     libatk-bridge2.0-0 \
+    libatk1.0-0 \
+    libc6 \
+    libcairo2 \
     libcups2 \
+    libdbus-1-3 \
+    libexpat1 \
+    libfontconfig1 \
+    libgbm1 \
+    libgcc1 \
+    libgdk-pixbuf2.0-0 \
+    libgl1 \
+    libgtk-3-0 \
+    libnspr4 \
+    libnss3 \
+    libpango-1.0-0 \
+    libpangocairo-1.0-0 \
+    libstdc++6 \
+    libx11-6 \
+    libx11-xcb1 \
+    libxcb1 \
     libxcomposite1 \
+    libxcursor1 \
     libxdamage1 \
+    libxext6 \
     libxfixes3 \
+    libxi6 \
     libxrandr2 \
-    libgbm1 \
-    libxkbcommon0 \
-    libasound2 \
-    libatspi2.0-0 && \
-    rm -rf /var/lib/apt/lists/*
-
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
-# Set up a new user named "user" with user ID 1000
-RUN useradd -m -u 1000 user
-
-# Set the working directory to the user's home directory
-WORKDIR /home/user/app
+    libxrender1 \
+    libxss1 \
+    libxtst6 \
+    lsb-release \
+    xdg-utils \
+    && rm -rf /var/lib/apt/lists/*
 
-# Switch to the "user" user
-USER user
+# Set the working directory
+WORKDIR /app
 
-# Set home and Playwright environment variables
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    PLAYWRIGHT_BROWSERS_PATH=/home/user/.cache/ms-playwright
+# Copy the requirements file
+COPY requirements.txt .
 
-# Explicitly create the Playwright cache directory with the correct user permissions
-RUN mkdir -p /home/user/.cache/ms-playwright
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the application code after switching to the user to maintain ownership
-COPY --chown=user . /home/user/app
+# Copy the application code
+COPY . .
 
-# Install Playwright and Chromium
-RUN pip install --no-cache-dir playwright && playwright install chromium
+# Install playwright browsers
+RUN playwright install --with-deps chromium
 
-CMD ["python", "app.py"]
+# Command to run the script
+CMD ["python", "app.py"]

app.py

@@ -2,12 +2,10 @@ import gradio as gr
 import os
 import selimium_try
 
-# Ensure a writable directory for Gradio flagging
-FLAGGING_DIR = "/home/user/app/flagged"
-os.makedirs(FLAGGING_DIR, exist_ok=True)
 
 def greet(name):
-    return selimium_try.main()
+    selimium_try.main()
+    return name
 
 def run():
     demo = gr.Interface(fn=greet, inputs="text", outputs="text", flagging_dir=FLAGGING_DIR)

requirements.txt

@@ -1 +1,3 @@
-gradio
+gradio
+playwright
+requests

selimium_try.py

@@ -1,82 +1,33 @@
 from playwright.sync_api import sync_playwright
-import json
-import time
+import sys
 
-COOKIE_FILE = "cookies.json"
-
-
-def load_cookies(context):
-    """Loads cookies from a file if available."""
-    try:
-        with open(COOKIE_FILE, "r") as f:
-            cookies = json.load(f)
-            context.add_cookies(cookies)
-            print("✅ Cookies loaded successfully.")
-    except FileNotFoundError:
-        print("⚠️ No cookies found. Logging in manually.")
-
-
-def save_cookies(context):
-    """Saves cookies to a file."""
-    cookies = context.cookies()
-    with open(COOKIE_FILE, "w") as f:
-        json.dump(cookies, f)
-        print("✅ Cookies saved successfully.")
-
-
-def is_logged_in_youtube(page):
-    """Checks if the user is logged into YouTube."""
-    page.goto("https://www.youtube.com", wait_until="domcontentloaded")
-    time.sleep(2)  # Wait for elements to load
-    try:
-        page.wait_for_selector("button#avatar-btn", timeout=5000)
-        print("✅ You are already logged in to YouTube!")
-        return True
-    except:
-        print("❌ You are NOT logged in to YouTube.")
-        return False
-
-
-def login_to_youtube(page):
-    """Logs into YouTube manually if needed."""
-    page.goto("https://accounts.google.com/signin/v2/identifier?service=youtube")
-
-    # Enter email
-    page.fill("input[type='email']", "your-email@gmail.com")
-    page.click("button:has-text('Next')")
-    time.sleep(3)
-
-    # Enter password
-    page.fill("input[type='password']", "your-password")  # ⚠️ Use a secure method for passwords
-    page.click("button:has-text('Next')")
-    time.sleep(5)
-
-    if is_logged_in_youtube(page):
-        print("✅ Login successful.")
-        return True
-    else:
-        print("❌ Login failed.")
-        return False
-
-def main():
+def search_youtube_title(search_term):
     with sync_playwright() as p:
-        browser = p.chromium.launch(headless=True)
-        context = browser.new_context()
+        browser = p.chromium.launch()
+        page = browser.new_page()
+        page.goto("https://www.youtube.com/")
 
-        # Load cookies before starting
-        load_cookies(context)
+        # Type the search term and press Enter
+        page.fill("input#search", search_term)
+        page.press("input#search", "Enter")
 
-        page = context.new_page()
-        result=is_logged_in_youtube(page)
-
-        # if not is_logged_in_youtube(page):
-            # if login_to_youtube(page):
-            #     save_cookies(context)
+        # Wait for the search results to load
+        page.wait_for_selector("#contents ytd-video-renderer")
 
+        # Extract the title of the first video
+        first_video_title = page.query_selector("#contents ytd-video-renderer #video-title").inner_text() if page.query_selector("#contents ytd-video-renderer #video-title") else "Title not found"
 
         browser.close()
-        return result
+        return first_video_title
+
+def main():
+    if len(sys.argv) > 1:
+        search_term = sys.argv[1]
+        title = search_youtube_title(search_term)
 
+        print(f"Title of the first video: {title}")
+    else:
+        print("Please provide a search term as a command-line argument.")
 
 if __name__ == "__main__":
     main()