Use login by supabase auth

app.py

@@ -1,8 +1,11 @@
 import os, time, re
 import gradio as gr
-
+import supabase
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+# Initialize the Supabase client
+client = supabase.create_client('https://zegvunzphdjucsvfkvky.supabase.co', 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InplZ3Z1bnpwaGRqdWNzdmZrdmt5Iiwicm9sZSI6InNlcnZpY2Vfcm9sZSIsImlhdCI6MTY4NTgzMzg4OCwiZXhwIjoyMDAxNDA5ODg4fQ.g6tVWYkncmCXwbnjtMzi0vR860wZtAUpqELBMkOYDaU')
+
 def inference(input_sentence, name_model):
     # Load model
     access_token = os.getenv("access_token")
@@ -51,22 +54,44 @@ def translator(input_sentence, name_model):
     
     return result
 
+def words(input_sentence):
+    len_inp = len(input_sentence.replace('\n', ' ').replace('  ', ' ').replace('  ', ' ').strip().split(' '))
+    return 'Words: ' + str(len_inp) + ' / 9999'
+    
 def login(username, password):
-    if username == os.getenv("username") and password == os.getenv("password"):
-        return True
+    # login
+    data = client.auth.sign_in_with_password({"email": username, "password": password})
+    return data
 
-if __name__ == "__main__":
-    MODELS = ['VuAI/khi-van', 'VuAI/vi2vi_vn98']
+MODELS = ['VuAI/khi-van', 'VuAI/vi2vi_vn98']
+    
+curr_words = 9999
+with gr.Blocks(title="Translator", theme=gr.themes.Default(primary_hue="red", secondary_hue="red")) as demo:
+    with gr.Row():
+        with gr.Column(scale=20):
+            gr.Text('VN98 Translation', show_label=False).style(container=False)
+        with gr.Column(scale=1, min_width=50):
+            darkBtn = gr.Button("Dark", variant='secondary', elem_id="darkTheme")
     
-    gr.Interface(fn=translator, 
-                inputs=[
-                    gr.inputs.Textbox(lines=5, label="Input Text"),
-                    gr.components.Dropdown(label="Select Model?", choices=MODELS, value=MODELS[0]),
-                ],
-                outputs=gr.inputs.Textbox(label="Result").style(show_copy_button=True),
-                title="Translation Demo",
-                examples = [
-                    ["Đây cũng là không có biện pháp sự tình, Vương Đằng Chi trước thì dùng thần hồn chi lực nhìn qua toàn bộ Thiên Thánh tông đệ tử, không có một cái có thể đạt tới Thái Sơ Thần Tông tuyển nhận tiêu chuẩn, mà lại Thiên Thánh tông đệ tử ưu tú thiên phú đều không khác mấy, không có đặc biệt xuất chúng tồn tại.",MODELS[0]],
-                    ['"Ba người các ngươi đi theo ta, ta trước mang các ngươi đi gặp Thái Sơ Thần Tông tông chủ đại nhân." Thiên Thánh tông tông chủ đối với trước mặt ba người nói.',MODELS[0]],
-                ]
-                ).launch(auth=login, auth_message="Chào mừng đến với VN98 Translator.")
+    with gr.Row():
+        with gr.Column():
+            inp = gr.Textbox(lines=5, label="Input Text")
+            inp.input(fn=words, inputs=inp, outputs=gr.Text(show_label=False, value='Words: 0 / ' + str(curr_words)))
+            sel = gr.Dropdown(label="Select Model?", choices=MODELS, value=MODELS[0])
+            btn = gr.Button("Run", variant='primary')
+        with gr.Column():
+            out = gr.Textbox(lines=5, label="Result", interactive=True).style(show_copy_button=True)
+
+    changeTheme = """function changeTheme() {
+        body = document.querySelector('body').classList.toggle('dark');
+        btn_theme = document.querySelector('#darkTheme');
+        btn_theme.innerText = (btn_theme.innerText == 'Light') ? 'Dark' : 'Light';
+    }
+    """
+
+    btn.click(translator, inputs=[inp, sel], outputs=[out])
+    darkBtn.click(fn=None, _js=changeTheme)
+
+
+if __name__ == "__main__":
+    demo.launch(share=True, auth=login, auth_message="Chào mừng đến với VN98 Translator.")