Login + File organization

.gitignore

@@ -83,5 +83,7 @@ http-client.private.env.json
 .idea/ApifoxUploaderProjectSetting.xml
 
 # Project specific
-key.json
+Resources/key.json
+Resources/salt.py
 venv
+**/__pycache__

.idea/vcs.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="VcsDirectoryMappings">
-    <mapping directory="$PROJECT_DIR$" vcs="" />
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
   </component>
 </project>

Resources/captions2.jsonl

@@ -0,0 +1,5 @@
+{"clean_text":"measure","start_time":0.0,"user_id":11,"signer":39,"file":"Aj9SDSAOXf4","end_time":1.435,"url":"www.youtube.com\/watch?v=Aj9SDSAOXf4","text":"test"}
+{"clean_text":"apple","start_time":1.5,"user_id":11,"signer":39,"file":"Aj9SDSAOXf4","end_time":1.6,"url":"www.youtube.com\/watch?v=Aj9SDSAOXf4","text":"hhhh"}
+{"clean_text":"test","start_time":null,"user_id":11,"signer":39,"file":"c2ORbHSQ5pw","end_time":null,"url":"www.youtube.com\/watch?v=c2ORbHSQ5pw","text":null}
+{"clean_text":"aaaaaaaaaa","start_time":null,"user_id":11,"signer":39,"file":"c2ORbHSQ5pw","end_time":null,"url":"www.youtube.com\/watch?v=c2ORbHSQ5pw","text":null}
+{"clean_text":"bbbbbbbb","start_time":null,"user_id":11,"signer":39,"file":"c2ORbHSQ5pw","end_time":null,"url":"www.youtube.com\/watch?v=c2ORbHSQ5pw","text":null}

Resources/css.py

@@ -0,0 +1,14 @@
+css = """
+.container {
+    position: relative;
+    width: 100%;
+    height: 0;
+    padding-bottom: 56.25%;
+}
+.video {
+    position: absolute;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+}"""

db_upload.py → Resources/db_upload.py

@@ -6,7 +6,11 @@ cred_obj = firebase_admin.credentials.Certificate('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")
-with open("videos.json", "r") as f:
+# videos_ref = db.reference("/Videos")
+# with open("videos.json", "r") as f:
+#     file_contents = json.load(f)
+# videos_ref.set(file_contents)
+users_ref = db.reference("/Users")
+with open("users.json", "r") as f:
     file_contents = json.load(f)
-videos_ref.set(file_contents)
+users_ref.set(file_contents)

Resources/users.json

@@ -0,0 +1,3 @@
+{
+  "sample_user": "f8f81086d3347f78c14e822641b1475da6f6921d5cb032acd400468aaca20f4d"
+}

main.py → app.py

@@ -1,58 +1,24 @@
 import gradio as gr
-import pandas as pd
-import re
 import firebase_admin
 from firebase_admin import db
-import hashlib
-import jsonlines
-import json
+from auth_functions import auth_function
+from video_player_functions import youtube_link_to_id, get_youtube_video
+from caption_editor_functions import get_captions, save, captions
+from Resources.css import css
 
-cred_obj = firebase_admin.credentials.Certificate('key.json')
+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")
 
-with open("captions.jsonl") as file:
-    captions = pd.read_json(file, lines=True)
-
-# videos = ["Aj9SDSAOXf4", "c2ORbHSQ5pw"]
-current_video = 0
 video_links = videos_ref.get()[1:]
-
-css = """
-.container {
-    position: relative;
-    width: 100%;
-    height: 0;
-    padding-bottom: 56.25%;
-}
-.video {
-    position: absolute;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-}"""
-
-
-def auth_function(username, password):
-    return True
-
-
-def youtube_link_to_id(link):
-    video_id = re.findall("=(.*?)&", link)
-    if len(video_id) == 0:
-        video_id = re.findall("=(.*)", link)
-    return video_id[0]
+current_video = 0
 
 
-def get_captions(video_id):
-    global captions
-    captions_edit = captions[captions['file'] == video_id]
-    captions_edit = captions_edit[['start_time', 'text', 'end_time']]
-    captions_edit.columns = ["Start", "Text", "End"]
-    return captions_edit
+def auth(username, password):
+    return auth_function(username, password, users_ref)
 
 
 def get_next_captions():
@@ -62,13 +28,6 @@ def get_next_captions():
     # return get_captions(videos[current_video])
 
 
-def get_youtube_video(video_id):
-    return f"""
-    <div class="container">
-    <iframe src="https://www.youtube.com/embed/{video_id}" frameborder="0" allowfullscreen class="video"></iframe>
-    </div>"""
-
-
 def get_next_youtube_video():
     global current_video, video_links
     current_video += 1
@@ -88,18 +47,6 @@ def refresh_components():
     return next_video, next_captions
 
 
-def save(df):
-    try:
-        global captions
-        captions['start_time'] = df['Start'].apply(lambda x: float(x))
-        captions['text'] = df['Text']
-        captions['end_time'] = df['End'].apply(lambda x: float(x))
-        captions.to_json('captions2.jsonl', orient='records', lines=True)
-        return "Save successful!"
-    except ValueError:
-        return "Save failed: Incorrect input format"
-
-
 with gr.Blocks(css=css) as app:
     gr.Markdown("## Caption Editor")
     with gr.Row():
@@ -118,4 +65,4 @@ with gr.Blocks(css=css) as app:
     next_video_button.click(fn=refresh_components, outputs=[video_embed, caption_editor])
     save_button.click(fn=save, inputs=caption_editor, outputs=save_result)
 
-app.launch(auth=auth_function)
+app.launch(auth=auth)

auth_functions.py

@@ -0,0 +1,17 @@
+from Resources.salt import salt
+import hashlib
+
+
+def encrypt(password):
+    result = hashlib.pbkdf2_hmac('sha256', bytes(password, "utf-8"), salt, 100000)
+    return result.hex()
+
+
+def auth_function(username, password, db_ref):
+    user_password = db_ref.child(username).get()
+    if user_password is None:
+        return False
+    pass_input = encrypt(password)
+    if user_password == pass_input:
+        return True
+    return False

caption_editor_functions.py

@@ -0,0 +1,25 @@
+import pandas as pd
+
+
+def get_captions(video_id):
+    global captions
+    captions_edit = captions[captions['file'] == video_id]
+    captions_edit = captions_edit[['start_time', 'text', 'end_time']]
+    captions_edit.columns = ["Start", "Text", "End"]
+    return captions_edit
+
+
+def save(df):
+    try:
+        global captions
+        captions['start_time'] = df['Start'].apply(lambda x: float(x))
+        captions['text'] = df['Text']
+        captions['end_time'] = df['End'].apply(lambda x: float(x))
+        captions.to_json('Resources/captions2.jsonl', orient='records', lines=True)
+        return "Save successful!"
+    except ValueError:
+        return "Save failed: Incorrect input format"
+
+
+with open("Resources/captions.jsonl") as file:
+    captions = pd.read_json(file, lines=True)

captions2.jsonl

@@ -1,2 +0,0 @@
-{"clean_text":"measure","start_time":0.0,"user_id":11,"signer":39,"file":"a1NeXe8bhuo","end_time":1.435,"url":"www.youtube.com\/watch?v=a1NeXe8bhuo","text":"measure"}
-{"clean_text":"apple","start_time":1.5,"user_id":11,"signer":39,"file":"a1NeXe8bhuo","end_time":1.6,"url":"www.youtube.com\/watch?v=a1NeXe8bhuo","text":"apple"}

video_player_functions.py

@@ -0,0 +1,15 @@
+import re
+
+
+def youtube_link_to_id(link):
+    video_id = re.findall("=(.*?)&", link)
+    if len(video_id) == 0:
+        video_id = re.findall("=(.*)", link)
+    return video_id[0]
+
+
+def get_youtube_video(video_id):
+    return f"""
+    <div class="container">
+    <iframe src="https://www.youtube.com/embed/{video_id}" frameborder="0" allowfullscreen class="video"></iframe>
+    </div>"""