Registration functionality

.idea/CaptionProject.iml

@@ -7,4 +7,11 @@
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
+  <component name="PyNamespacePackagesService">
+    <option name="namespacePackageFolders">
+      <list>
+        <option value="$MODULE_DIR$/Pages" />
+      </list>
+    </option>
+  </component>
 </module>

Pages/Functions/auth_functions.py

@@ -1,6 +1,6 @@
 from ..Resources.salt import salt
 import hashlib
-from db_connection import users_ref
+from .db_connection import users_ref, get_user_ref
 
 
 def encrypt(password):
@@ -8,11 +8,26 @@ def encrypt(password):
     return result.hex()
 
 
-def auth_function(username, password):
+def get_password_by_username(username):
     user_password = users_ref.child(username).get()
+    return user_password
+
+
+def auth_function(username, password):
+    user_password = get_password_by_username(username)
     if user_password is None:
         return False
     pass_input = encrypt(password)
     if user_password == pass_input:
         return True
     return False
+
+
+def register(username, password):
+    if get_password_by_username(username) is not None:
+        return "Registration error: Username already exists"
+    user_ref = get_user_ref(username)
+    pass_hash = encrypt(password)
+    user_ref.set(pass_hash)
+    return "Registration successful!"
+

Pages/Functions/caption_editor_functions.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import json
 
-FILE_PATH = "../Resources/captions.jsonl"
+FILE_PATH = "Pages/Resources/captions.jsonl"
 
 
 def get_captions_by_video_id(video_id):

Pages/Functions/db_connection.py

@@ -1,7 +1,7 @@
 import firebase_admin
 from firebase_admin import db
 
-KEY_PATH = '../Resources/key.json'
+KEY_PATH = 'Pages/Resources/key.json'
 
 cred_obj = firebase_admin.credentials.Certificate(KEY_PATH)
 default_app = firebase_admin.initialize_app(cred_obj, {
@@ -9,3 +9,7 @@ default_app = firebase_admin.initialize_app(cred_obj, {
     })
 videos_ref = db.reference("/Videos")
 users_ref = db.reference("/Users")
+
+
+def get_user_ref(user):
+    return db.reference("/Users/" + user)

Pages/Functions/video_player_functions.py

@@ -1,5 +1,5 @@
 import re
-from db_connection import videos_ref
+from .db_connection import videos_ref
 
 
 def youtube_link_to_id(link):

Pages/main_page.py

@@ -1,7 +1,7 @@
 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
+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 = ""

Pages/registration_page.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from .Functions.auth_functions import register
 
 with gr.Blocks() as registration:
     with gr.Column(variant="panel"):
@@ -8,4 +9,4 @@ with gr.Blocks() as registration:
         registration_button = gr.Button(value="Register")
         registration_result = gr.Markdown()
 
-        # registration_button.click(fn=func, inputs=[username, password], outputs=registration_result)
+        registration_button.click(fn=register, inputs=[username, password], outputs=registration_result)