Update main.py

main.py

@@ -1,7 +1,6 @@
 import os
 import json
 import bcrypt
-import js2py
 from typing import List
 from pathlib import Path
 from langchain_huggingface import HuggingFaceEmbeddings
@@ -27,6 +26,7 @@ from chainlit.playground.config import add_llm_provider
 from chainlit.playground.providers.langchain import LangchainGenericProvider
 
 from deep_translator import GoogleTranslator
+from browser import document, html
 
 @cl.password_auth_callback
 def auth_callback(username: str, password: str):
@@ -200,31 +200,10 @@ async def on_message(message: cl.Message):
                 )
         
     async with cl.Step(type="run", name="Réponse de Mistral"):
-        Display = js2py.eval_js6('''
-            class Display {
-                constructor(name) {
-                    this.name = name;
-                }
-
-                doSomething() {
-                    var parentElement = document.getElementsByClassName('css-oa138a');
-                    var theFirstChild = parentElement.firstChild;
-                    var newElement = document.createElement('div');
-                    newElement.setAttribute('class', 'MuiPaper-root MuiPaper-elevation MuiPaper-rounded MuiPaper-elevation0 MuiAlert-root MuiAlert-standardInfo MuiAlert-standard css-179rm25');
-                    newElement.setAttribute('role', 'info');
-                    newElement.innerHTML = '<svg class="MuiSvgIcon-root MuiSvgIcon-fontSizeInherit css-1cw4hi4" focusable="false" aria-hidden="true" viewBox="0 0 24 24" data-testid="InfoOutlinedIcon"><path d="M11,9H13V7H11M12,20C7.59,20 4,16.41 4,12C4,7.59 7.59,4 12,4C16.41,4 20,7.59 20, 12C20,16.41 16.41,20 12,20M12,2A10,10 0 0,0 2,12A10,10 0 0,0 12,22A10,10 0 0,0 22,12A10, 10 0 0,0 12,2M11,17H13V11H11V17Z"></path></svg>' + this.name;
-                    parentElement.insertBefore(newElement, theFirstChild);
-                    console.log(this.name);
-                }
-            }''')
-        Display.new(str(message.content)).doSomething()
-        
-        #display = js2py.eval_js("""
-        #    function displayAnswer(response){
-        #        return console.log(response);
-        #    }""")
-        #response = str(message.content)
-        #displayAnswer(response)
+        response = str(message.content)
+        parentElement = document.getElementsByClassName("css-oa138a")
+        theFirstChild = parentElement.get(selector="div")[0]
+        parentElement.insertBefore(html.DIV(f"{response}",Class="MuiPaper-root MuiPaper-elevation MuiPaper-rounded MuiPaper-elevation0 MuiAlert-root MuiAlert-standardInfo MuiAlert-standard css-179rm25"), theFirstChild)
         
         cb = cl.AsyncLangchainCallbackHandler()
         with tracing_v2_enabled():