add playwright

Dockerfile

@@ -9,6 +9,9 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+RUN python -m playwright install-deps
+RUN playwright install webkit
+
 RUN wget -O - https://www.openssl.org/source/openssl-1.1.1u.tar.gz | tar zxf - \
     && cd openssl-1.1.1u \
     && ./config --prefix=/usr/local \

app.py

@@ -62,6 +62,48 @@ from langchain.document_loaders import (
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.docstore.document import Document
 import langchain
+import asyncio
+from playwright.async_api import async_playwright
+
+async def start_playwright():
+    pw = await async_playwright().start()
+    # browser = await pw.chromium.launch(headless=True)
+    browser = await pw.webkit.launch(headless=True)
+    page = await browser.new_page()
+
+    # note all methods are async (use the "await" keyword)
+    await page.goto("https://basicchatbot.azurewebsites.net/")
+    print("Title of Web: ", await page.title())
+    await page.wait_for_timeout(200)
+#     print("Content of Web: ", await page.content())
+#     print("Test content: ", await page.locator("//div[@class='css-zt5igj e1nzilvr3']").inner_html())
+    print("Test content: ", await page.locator("//div[@class='css-zt5igj e1nzilvr3']").inner_text())
+    
+    await page.locator("//textarea").fill("when memory is not enough")
+    await page.wait_for_timeout(200)
+#     print("Content of Web: ", await page.content())
+#     await page.locator("//button[@class='css-1wi2cd3 e1d2x3se3']").click()
+    await page.locator("//textarea").press("Enter")
+    await page.wait_for_timeout(3000)
+#     print("Content of Web: ", await page.content())
+#     print("output_text 1", await page.locator("//div[@aria-label='Chat message from assistant']").last.inner_text())
+#     output_text = await page.locator("//div[@aria-label='Chat message from assistant']").last.inner_text()
+#     print("output_text 1", output_text)
+    output_history = "NOTHING"
+    for i in range(40):
+        output_text = await page.locator("//div[@aria-label='Chat message from assistant']").last.inner_text()
+        print("output_text... :", output_text)
+        if output_text == output_history:
+            break
+        else:
+            await page.wait_for_timeout(1000)
+            output_history = output_text
+    
+    print("-------- Final Answer-----------\n", output_text)
+    
+    await browser.close()
+
+
 # import playsound
 
 langchain.debug = True
@@ -1232,10 +1274,12 @@ if __name__ == '__main__':
     # QAQuery("what is COFOR ?")
     # CreatDb_P()
     # QAQuery_p("what is GST ?")
+    asyncio.run(start_playwright())
     if SysLock == "1":
         demo.queue().launch(auth=(username, password), server_name="0.0.0.0", server_port=7860)
     else:
         demo.queue().launch(server_name="0.0.0.0", server_port=7860)
     pass
+    
 
 

requirements.txt

@@ -13,4 +13,5 @@ playsound
 pygame
 unstructured
 openpyxl
-python-docx
+python-docx
+playwright