Update app.py

app.py

@@ -1,201 +1,48 @@
-import asyncio
-import json
-import os  # Import the os module
-from playwright.async_api import async_playwright
-
-
-class HuggingChatAPI:
-    def __init__(self, headless=True):
-        self.username = os.environ.get("HUGGINGFACE_USERNAME")
-        self.password = os.environ.get("HUGGINGFACE_PASSWORD")
-        if not self.username or not self.password:
-            raise ValueError("HUGGINGFACE_USERNAME and HUGGINGFACE_PASSWORD environment variables must be set.")
-
-        self.headless = headless
-        self.playwright = None
-        self.browser = None
-        self.page = None
-        self.current_model = None
-        self.conversation_history = []  # Store conversation history
-
-    async def start(self):
-        self.playwright = await async_playwright().start()
-        self.browser = await self.playwright.chromium.launch(headless=self.headless)
-        self.page = await self.browser.new_page()
-        await self.page.set_viewport_size({"width": 1161, "height": 813})
-        await self.login()
-
-    async def stop(self):
-        if self.browser:
-            await self.browser.close()
-        if self.playwright:
-            await self.playwright.stop()
-
-    async def login(self):
-        await self.page.goto("https://huggingface.co/chat/")
-        await self.page.locator('body > div:nth-of-type(2) span').click()  # "with Hugging Face" button
-        await self.page.locator('::-p-aria(Sign in  with Hugging Face) >>>> ::-p-aria([role="image"])').click()
-        await self.page.locator('::-p-aria(Username or Email address)').fill(self.username)
-        await self.page.locator('::-p-aria(Password)').fill(self.password)
-        await self.page.locator('::-p-aria(Login)').click()
-        await self.page.wait_for_url("https://huggingface.co/chat/")  # Wait for successful login
-
-
-    async def select_model(self, model_name):
-        """Selects a model by name (partial matching)."""
-
-        await self.page.locator('::-p-aria(Models 11)').click()
-
-        model_locator = None
-        if model_name == "meta-llama/Llama-3.3-70B-Instruct":
-            model_locator = self.page.locator('::-p-aria(View details for meta-llama/Llama-3.3-70B-Instruct)')
-        elif model_name == "Qwen/Qwen2.5-72B-Instruct":
-            model_locator = self.page.locator('::-p-aria(View details for Qwen/Qwen2.5-72B-Instruct)')
-        elif model_name == "CohereForAI/c4ai-command-r-plus-08-2024":
-            model_locator = self.page.locator('::-p-aria(View details for CohereForAI/c4ai-command-r-plus-08-2024)')
-        elif model_name == "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B":
-             model_locator = self.page.locator('::-p-aria(View details for deepseek-ai/DeepSeek-R1-Distill-Qwen-32B)')
-        elif model_name == "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF":
-            model_locator = self.page.locator('::-p-aria(View details for nvidia/Llama-3.1-Nemotron-70B-Instruct-HF)')
-        elif model_name == "Qwen/QwQ-32B-Preview":
-            model_locator = self.page.locator('::-p-aria(View details for Qwen/QwQ-32B-Preview)')
-        elif model_name == "Qwen/Qwen2.5-Coder-32B-Instruct":
-            model_locator = self.page.locator('::-p-aria(View details for Qwen/Qwen2.5-Coder-32B-Instruct)')
-        elif model_name == "meta-llama/Llama-3.2-11B-Vision-Instruct":
-            model_locator = self.page.locator('::-p-aria(View details for meta-llama/Llama-3.2-11B-Vision-Instruct)')
-        elif model_name == "NousResearch/Hermes-3-Llama-3.1-8B":
-            model_locator = self.page.locator('::-p-aria(View details for NousResearch/Hermes-3-Llama-3.1-8B)')
-        elif model_name == "mistralai/Mistral-Nemo-Instruct-2407":
-            model_locator = self.page.locator('::-p-aria(View details for mistralai/Mistral-Nemo-Instruct-2407)')
-        elif model_name == "microsoft/Phi-3.5-mini-instruct":
-            model_locator = self.page.locator('::-p-aria(View details for microsoft/Phi-3.5-mini-instruct)')
-        else:
-            raise ValueError(f"Model '{model_name}' not found in the predefined list.")
-
-
-        if model_locator:
-            await model_locator.click()
-            self.current_model = model_name
-        else:
-            raise ValueError(f"Model '{model_name}' selection failed.")
-        
-    async def set_system_prompt(self, system_prompt):
-        """Sets the system prompt."""
-        await self.page.locator("div.p-3 path").click()
-        await self.page.locator('::-p-aria(Custom system prompt)').fill(system_prompt)
-        await self.page.locator('::-p-aria(New chat)').click()
-
-
-    async def toggle_web_search(self, enable: bool):
-      """Enable or disable web search."""
-      await self.page.locator('::-p-aria(Search)').click() #open tools
-      code_executor_button = self.page.locator('::-p-aria(Code Executor)')
-
-      if enable:
-            # Check if already enabled
-            if "bg-gray-100" not in await code_executor_button.get_attribute("class"):
-                await code_executor_button.click()  # Enable
-      else:
-            # Check if already disabled
-            if "bg-gray-100" in await code_executor_button.get_attribute("class"):
-                await code_executor_button.click() #click to open
-                await self.page.locator('::-p-aria(Deactivate)').click() # Disable
-      await self.page.locator('div:nth-of-type(2) > div > div > button path').click() #close tools
-
-
-
-    async def send_message(self, message: str, expect_response: bool = True, timeout: float = 60.0):
-        """Sends a message and optionally waits for a response."""
-
-        self.conversation_history.append({"role": "user", "content": message})
-        await self.page.locator('::-p-aria(Ask anything)').fill(message)
-        await self.page.locator('::-p-aria(Send message)').click()
-
-        if expect_response:
-            try:
-                # Wait for a new message bubble to appear.  This is a more robust way
-                # to detect a response than just waiting a fixed amount of time.
-                await self.page.locator('.group.self-end').last.wait_for(timeout=timeout * 1000)  # milliseconds
-                response_elements = await self.page.locator('.group.self-end > .markdown').all()
-                response = await response_elements[-1].inner_text() # Get the last response
-                self.conversation_history.append({"role": "assistant", "content": response})
-                return response
-
-            except Exception as e:
-                print(f"Error waiting for response: {e}")
-                return None  # Or raise the exception, depending on your needs
-        else:
-            return None
-
-
-
-    async def new_chat(self):
-        """Starts a new chat."""
-        await self.page.locator('::-p-aria(New chat)').click()
-        self.conversation_history = []  # Clear history for the new chat
-
-
-    async def chat(self, message: str, model_name: str = None, system_prompt: str = None, web_search: bool = False, timeout: float = 60.0):
-        """
-        Combined method for setting parameters and sending a message.
-
-        Args:
-            message: The user's message.
-            model_name:  The name of the model to use.
-            system_prompt: The system prompt to set.
-            web_search: Whether to enable web search.
-            timeout: Timeout for waiting for a response, in seconds.
-
-        Returns:
-            The model's response, or None if no response is received.
-        """
-        if model_name and model_name != self.current_model:
-            await self.new_chat()  # Start a new chat when changing models
-            await self.select_model(model_name)
-
-        if system_prompt is not None:  # Allow empty string "" as a valid prompt
-            await self.set_system_prompt(system_prompt)
-
-        if web_search is not None:
-            await self.toggle_web_search(web_search)
-
-        return await self.send_message(message, timeout=timeout)
-
-
-async def main():
-    # --- Example Usage ---
-    chat_api = HuggingChatAPI(headless=False)  # Set headless=False for debugging
-    try:
-        await chat_api.start()
-
-        # Example 1: Simple chat with default model
-        response1 = await chat_api.chat("What is the capital of France?")
-        print(f"Response 1: {response1}")
-
-        # Example 2: Chat with a specific model and system prompt
-        response2 = await chat_api.chat("Explain the theory of relativity.",
-                                        model_name="Qwen/Qwen2.5-72B-Instruct",
-                                        system_prompt="You are a helpful and concise assistant.")
-        print(f"Response 2: {response2}")
-
-        # Example 3:  Enable web search
-        response3 = await chat_api.chat("What is the latest news on AI?",
-                                        model_name="CohereForAI/c4ai-command-r-plus-08-2024",
-                                        web_search=True)
-        print(f"Response 3: {response3}")
-
-        # Example 4: Continue the first conversation
-        response4 = await chat_api.chat("And what about Germany?")  # No model/prompt change
-        print(f"Response 4: {response4}")
-
-        # Example 5: Get conversation history
-        print("\nConversation History:")
-        for message in chat_api.conversation_history:
-            print(f"- {message['role']}: {message['content']}")
-
-
-    finally:
-        await chat_api.stop()
-
-if __name__ == "__main__":
-    asyncio.run(main())
+FROM python:3.11-slim AS base
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    wget \
+    unzip \
+    libx11-xcb1 \
+    libxcomposite1 \
+    libxcursor1 \
+    libxdamage1 \
+    libxi6 \
+    libxtst6 \
+    libnss3 \
+    libcups2 \
+    libxss1 \
+    libxrandr2 \
+    libasound2 \
+    libatk1.0-0 \
+    libatk-bridge2.0-0 \
+    libcairo2 \
+    libgdk-pixbuf2.0-0 \
+    libgtk-3-0 \
+    libpango-1.0-0 \
+    libpangocairo-1.0-0 \
+    libx11-6 \
+    xvfb \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY . /app/
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+ENV PLAYWRIGHT_BROWSERS_PATH=/ms-playwright
+RUN mkdir -p $PLAYWRIGHT_BROWSERS_PATH && playwright install chromium --with-deps
+
+RUN useradd -m -u 1000 user
+RUN mkdir /data
+RUN chown -R 1000:1000 /app /data
+
+COPY entrypoint.sh /app/entrypoint.sh
+RUN chmod +x /app/entrypoint.sh
+
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+ENTRYPOINT ["xvfb-run", "/app/entrypoint.sh"]