Mounted as Fastapi app: Added multipage support

.gitignore

@@ -83,11 +83,11 @@ http-client.private.env.json
 .idea/ApifoxUploaderProjectSetting.xml
 
 # Manually added
-Resources/key.json
-Resources/salt.py
-Resources/upload.py
-Resources/users.json
-Resources/videos.json
-Resources/captions_backup.jsonl
+Pages/Resources/key.json
+Pages/Resources/salt.py
+Pages/Resources/upload.py
+Pages/Resources/users.json
+Pages/Resources/videos.json
+Pages/Resources/captions_backup.jsonl
 venv
 **/__pycache__

auth_functions.py → Pages/Functions/auth_functions.py

@@ -1,5 +1,6 @@
-from Resources.salt import salt
+from ..Resources.salt import salt
 import hashlib
+from db_connection import users_ref
 
 
 def encrypt(password):
@@ -7,8 +8,8 @@ def encrypt(password):
     return result.hex()
 
 
-def auth_function(username, password, db_ref):
-    user_password = db_ref.child(username).get()
+def auth_function(username, password):
+    user_password = users_ref.child(username).get()
     if user_password is None:
         return False
     pass_input = encrypt(password)

caption_editor_functions.py → Pages/Functions/caption_editor_functions.py

@@ -1,9 +1,11 @@
 import pandas as pd
 import json
 
+FILE_PATH = "../Resources/captions.jsonl"
+
 
 def get_captions_by_video_id(video_id):
-    with open("Resources/captions.jsonl") as file:
+    with open(FILE_PATH) as file:
         captions = pd.read_json(file, lines=True)
 
     captions_edit = captions[captions['file'] == video_id]
@@ -16,9 +18,8 @@ def save_dataframe(df, video_id, user):
     cols = ["clean_text", "start_time", "user_id", "signer", "file", "end_time", "url", "text"]
     other_captions_data = []
     new_captions_data = []
-    file_name = "Resources/captions.jsonl"
 
-    with open(file_name) as f:
+    with open(FILE_PATH) as f:
         for line in f:
             caption = json.loads(line)
             if caption['file'] == video_id:
@@ -36,7 +37,7 @@ def save_dataframe(df, video_id, user):
 
         all_captions = pd.concat([other_captions, new_captions], ignore_index=True)
 
-        all_captions.to_json('Resources/captions.jsonl', orient='records', lines=True)
+        all_captions.to_json(FILE_PATH, orient='records', lines=True)
         return "Save successful!"
     except ValueError:
         return "Save failed: Incorrect input format"

Pages/Functions/db_connection.py

@@ -0,0 +1,11 @@
+import firebase_admin
+from firebase_admin import db
+
+KEY_PATH = '../Resources/key.json'
+
+cred_obj = firebase_admin.credentials.Certificate(KEY_PATH)
+default_app = firebase_admin.initialize_app(cred_obj, {
+    'databaseURL': "https://video-link-db-default-rtdb.europe-west1.firebasedatabase.app/"
+    })
+videos_ref = db.reference("/Videos")
+users_ref = db.reference("/Users")

video_player_functions.py → Pages/Functions/video_player_functions.py

@@ -1,4 +1,5 @@
 import re
+from db_connection import videos_ref
 
 
 def youtube_link_to_id(link):
@@ -15,6 +16,6 @@ def get_video_embed_by_id(video_id):
     </div>"""
 
 
-def get_video_link_by_pointer(pointer, db_ref):
-    video_link = db_ref.child(str(pointer)).get()
+def get_video_link_by_pointer(pointer):
+    video_link = videos_ref.child(str(pointer)).get()
     return video_link

{Resources → Pages/Resources}/captions.jsonl

@@ -1,6 +1,4 @@
 {"clean_text":"one","start_time":1.5,"user_id":"a","signer":40,"file":"Y4stD_ypaAI","end_time":1.6,"url":"www.youtube.com\/watch?v=Y4stD_ypaAI","text":"one"}
-{"clean_text":"measure","start_time":0.0,"user_id":"admin","signer":20,"file":"tkMg8g8vVUo","end_time":1.435,"url":"www.youtube.com\/watch?v=tkMg8g8vVUo","text":"measure"}
-{"clean_text":"apple","start_time":1.7,"user_id":"admin","signer":20,"file":"tkMg8g8vVUo","end_time":2.0,"url":"www.youtube.com\/watch?v=tkMg8g8vVUo","text":"apple"}
 {"clean_text":"test","start_time":0.0,"user_id":"admin","signer":39,"file":"1pXUgdCnVec","end_time":1.435,"url":"www.youtube.com\/watch?v=1pXUgdCnVec","text":"test"}
 {"clean_text":"test","start_time":1.5,"user_id":"admin","signer":39,"file":"1pXUgdCnVec","end_time":1.6,"url":"www.youtube.com\/watch?v=1pXUgdCnVec","text":"test 2"}
 {"clean_text":"test","start_time":0.0,"user_id":"admin","signer":39,"file":"1pXUgdCnVec","end_time":1.435,"url":"www.youtube.com\/watch?v=1pXUgdCnVec","text":"test"}
@@ -13,3 +11,5 @@
 {"clean_text":"test","start_time":0.0,"user_id":"admin","signer":39,"file":"1pXUgdCnVec","end_time":1.435,"url":"www.youtube.com\/watch?v=1pXUgdCnVec","text":"test"}
 {"clean_text":"test","start_time":0.0,"user_id":"admin","signer":39,"file":"1pXUgdCnVec","end_time":1.435,"url":"www.youtube.com\/watch?v=1pXUgdCnVec","text":"test"}
 {"clean_text":"test","start_time":0.0,"user_id":"admin","signer":39,"file":"1pXUgdCnVec","end_time":1.435,"url":"www.youtube.com\/watch?v=1pXUgdCnVec","text":"test"}
+{"clean_text":"measure","start_time":0.0,"user_id":"admin","signer":20,"file":"tkMg8g8vVUo","end_time":1.435,"url":"www.youtube.com\/watch?v=tkMg8g8vVUo","text":"measure"}
+{"clean_text":"apple","start_time":1.71,"user_id":"admin","signer":20,"file":"tkMg8g8vVUo","end_time":2.0,"url":"www.youtube.com\/watch?v=tkMg8g8vVUo","text":"apple"}

Pages/main_page.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from Functions.video_player_functions import youtube_link_to_id, get_video_embed_by_id, get_video_link_by_pointer
+from Functions.caption_editor_functions import get_captions_by_video_id, save_dataframe
+from Resources.css import css
+
+next_video_pointer = 0
+user = ""
+
+
+def get_username(request: gr.Request):
+    global user
+    user = request.username
+
+
+def save(df, video_id):
+    return save_dataframe(df, video_id, user)
+
+
+def get_next_components():
+    global next_video_pointer
+    next_video_link = get_video_link_by_pointer(next_video_pointer)
+    next_video_pointer += 1
+    if next_video_link is None:
+        next_video_link = get_video_link_by_pointer(0)
+        next_video_pointer = 1
+
+    next_video_id = youtube_link_to_id(next_video_link)
+
+    next_video = get_video_embed_by_id(next_video_id)
+    next_captions = get_captions_by_video_id(next_video_id)
+
+    return next_video, next_captions, next_video_id
+
+
+(start_video, start_captions, start_video_id) = get_next_components()
+
+with gr.Blocks(css=css) as main_page:
+    gr.Markdown("## Caption Editor")
+    current_video_id = gr.Textbox(value=start_video_id, visible=False, interactive=False)
+    with gr.Row():
+        with gr.Column():
+            caption_editor = gr.DataFrame(interactive=True,
+                                          value=start_captions,
+                                          datatype=["number", "str", "number"],
+                                          row_count=(start_captions.shape[0], "fixed"),
+                                          col_count=(3, "fixed"), column_widths=["20%", "60%", "20%"])
+            save_button = gr.Button(value="Save")
+            save_result = gr.Markdown()
+        with gr.Column():
+            video_embed = gr.HTML(value=start_video)
+            next_video_button = gr.Button("Next")
+
+    next_video_button.click(fn=get_next_components,
+                            outputs=[video_embed, caption_editor, current_video_id])
+    save_button.click(fn=save,
+                      inputs=[caption_editor, current_video_id],
+                      outputs=save_result)
+    main_page.load(get_username)

Pages/registration_page.py

@@ -0,0 +1,11 @@
+import gradio as gr
+
+with gr.Blocks() as registration:
+    with gr.Column(variant="panel"):
+        gr.Markdown(value="# Create Account")
+        username = gr.Textbox(label="Username", show_label=True, lines=1)
+        password = gr.Textbox(label="Password", show_label=True, lines=1, type="password")
+        registration_button = gr.Button(value="Register")
+        registration_result = gr.Markdown()
+
+        # registration_button.click(fn=func, inputs=[username, password], outputs=registration_result)

app.py

@@ -1,74 +1,25 @@
+from fastapi import FastAPI
+from fastapi.responses import HTMLResponse
 import gradio as gr
-import firebase_admin
-from firebase_admin import db
-from auth_functions import auth_function
-from video_player_functions import youtube_link_to_id, get_video_embed_by_id, get_video_link_by_pointer
-from caption_editor_functions import get_captions_by_video_id, save_dataframe
-from Resources.css import css
+from Pages.main_page import main_page
+from Pages.Functions.auth_functions import auth_function
+from Pages.registration_page import registration
 
-cred_obj = firebase_admin.credentials.Certificate('Resources/key.json')
-default_app = firebase_admin.initialize_app(cred_obj, {
-    'databaseURL': "https://video-link-db-default-rtdb.europe-west1.firebasedatabase.app/"
-    })
-videos_ref = db.reference("/Videos")
-users_ref = db.reference("/Users")
+app = FastAPI()
+MAIN_PATH = "/caption_editor"
+REGISTRATION_PATH = "/registration"
 
-next_video_pointer = 0
-user = ""
+index_html = f'''
+<div>
+<iframe src={MAIN_PATH} width=100% height=100% frameBorder="0"></iframe>
+</div>
+'''
 
 
-def auth(username, password):
-    return auth_function(username, password, db_ref=users_ref)
+@app.get("/", response_class=HTMLResponse)
+def index():
+    return index_html
 
 
-def get_username(request: gr.Request):
-    global user
-    user = request.username
-
-
-def save(df, video_id):
-    return save_dataframe(df, video_id, user)
-
-
-def get_next_components():
-    global next_video_pointer
-    next_video_link = get_video_link_by_pointer(next_video_pointer, db_ref=videos_ref)
-    next_video_pointer += 1
-    if next_video_link is None:
-        next_video_link = get_video_link_by_pointer(0, db_ref=videos_ref)
-        next_video_pointer = 1
-
-    next_video_id = youtube_link_to_id(next_video_link)
-
-    next_video = get_video_embed_by_id(next_video_id)
-    next_captions = get_captions_by_video_id(next_video_id)
-
-    return next_video, next_captions, next_video_id
-
-
-(start_video, start_captions, start_video_id) = get_next_components()
-
-with gr.Blocks(css=css) as app:
-    gr.Markdown("## Caption Editor")
-    current_video_id = gr.Textbox(value=start_video_id, visible=False, interactive=False)
-    with gr.Row():
-        with gr.Column():
-            caption_editor = gr.DataFrame(interactive=True,
-                                          value=start_captions,
-                                          datatype=["number", "str", "number"],
-                                          row_count=(start_captions.shape[0], "fixed"),
-                                          col_count=(3, "fixed"), column_widths=["20%", "60%", "20%"])
-            save_button = gr.Button(value="Save")
-            save_result = gr.Markdown()
-        with gr.Column():
-            video_embed = gr.HTML(value=start_video)
-            next_video_button = gr.Button("Next")
-
-    next_video_button.click(fn=get_next_components,
-                            outputs=[video_embed, caption_editor, current_video_id])
-    save_button.click(fn=save,
-                      inputs=[caption_editor, current_video_id],
-                      outputs=save_result)
-    app.load(get_username)
-
-app.launch(auth=auth)
+app = gr.mount_gradio_app(app, main_page, path=MAIN_PATH, auth=auth_function)
+app = gr.mount_gradio_app(app, registration, path=REGISTRATION_PATH)