Use the GPM package + draft the UI

README.md

@@ -13,8 +13,9 @@ short_description: 'Stateless password manager, powered by ML neural networks.'
 
 ## GPM: Generative Password Manager <img src="images/logo.png" alt="apehex logo" width="32" height="32">
 
-Stateless password manager, powered by ML neural networks.
+[Stateless password manager][github-gpm], powered by ML neural networks.
 
 Licensed under the [aGPLv3][github-license].
 
+[github-gpm]: https://github.com/apehex/gpm
 [github-license]: LICENSE.md

app.py

@@ -1,7 +1,160 @@
-import gradio as gr
+import functools
 
-def greet(name):
-    return "Hello " + name + "!!"
+import gradio
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+import gpm.pipeline
+
+# META #########################################################################
+
+STYLE = '''.white-text span { color: white; }'''
+TITLE = '''Generative Password Manager'''
+INTRO = '''This is a POC, do **not** use it to manage your secrets.\nStateless password manager: you don't need to save passwords, they can all be derived from a single master key.\nAlways use the same format for a given target / ID: for example the password generated for "Github" and "github.com" are different.'''
+
+# ENUMS ########################################################################
+
+# password level
+CHARS = 0
+WORDS = 1
+
+# password alphabet
+DIGITS = 1
+LOWERS = 2
+UPPERS = 4
+SPACES = 8
+SYMBOLS = 16
+
+# INTRO ########################################################################
+
+def create_intro_block(intro: str) -> dict:
+    __intro = gradio.Markdown(intro, line_breaks=True)
+    return {'intro_block': __intro}
+
+# MASTER #######################################################################
+
+def create_master_block() -> dict:
+    __key = gradio.Textbox(label='Key', type='text', value='', placeholder='Your master key.', lines=1, max_lines=1, scale=1, show_copy_button=True, interactive=True)
+    return {
+        'key_block': __key,}
+
+# VOCABULARY ###################################################################
+
+def create_vocabulary_block() -> dict:
+    __level = gradio.Radio(label='Level', type='value', value=CHARS, choices=[('Character', CHARS), ('Word', WORDS)], interactive=True)
+    __vocabulary = gradio.CheckboxGroup(label='Vocabulary', type='value', value=[DIGITS, LOWERS, UPPERS], choices=[('Digits', DIGITS), ('Lowercase', LOWERS), ('Uppercase', UPPERS), ('Spaces', SPACES), ('Symbols', SYMBOLS)], interactive=True)
+    return {
+        'level_block': __level,
+        'vocabulary_block': __vocabulary,}
+
+# SAMPLING #####################################################################
+
+def create_sampling_block() -> dict:
+    __length = gradio.Slider(label='Length', value=8, minimum=1, maximum=32, step=1, scale=1, interactive=True)
+    __nonce = gradio.Number(label='Nonce', value=1, minimum=0, maximum=2 ** 32, step=1, scale=1, interactive=True)
+    return {
+        'length_block': __length,
+        'nonce_block': __nonce,}
+
+# INPUTS #######################################################################
+
+def create_inputs_block() -> dict:
+    __target = gradio.Textbox(label='Target', type='text', value='', placeholder='The login target (URL, IP, name, etc), like "Hugging Face" or "https://github.com".', lines=1, max_lines=1, scale=1, show_copy_button=True, interactive=True)
+    __identifier = gradio.Textbox(label='Identifier', type='text', value='', placeholder='The login ID (username, email, etc), like "John Doe" or "john.doe@example.com".', lines=1, max_lines=1, scale=1, show_copy_button=True, interactive=True)
+    return {
+        'target_block': __target,
+        'identifier_block': __identifier,}
+
+# OUTPUTS ######################################################################
+
+def create_outputs_block() -> dict:
+    __password = gradio.Textbox(label='Password', type='text', value='', placeholder='The generated password.', lines=1, max_lines=1, scale=1, show_copy_button=True, interactive=False)
+    return {
+        'password_block': __password,}
+
+# ACTIONS ######################################################################
+
+def create_actions_block() -> dict:
+    __process = gradio.Button('Generate', variant='primary', size='lg', scale=1, interactive=True)
+    return {'process_block': __process,}
+
+# STATE ########################################################################
+
+def create_state() -> dict:
+    return {}
+
+# LAYOUT #######################################################################
+
+def create_layout(intro: str=INTRO) -> dict:
+    __fields = {}
+    __fields.update(create_intro_block(intro=intro))
+    with gradio.Tabs():
+        with gradio.Tab('Manager') as __main_tab:
+            __fields.update({'main_tab': __main_tab})
+            with gradio.Row(equal_height=True):
+                __fields.update(create_inputs_block())
+            with gradio.Row(equal_height=True):
+                __fields.update(create_outputs_block())
+            with gradio.Row(equal_height=True):
+                __fields.update(create_actions_block())
+        with gradio.Tab('Settings') as __settings_tab:
+            __fields.update({'settings_tab': __settings_tab})
+            with gradio.Column(scale=1):
+                with gradio.Row(equal_height=True):
+                    __fields.update(create_master_block())
+                with gradio.Row(equal_height=True):
+                    __fields.update(create_vocabulary_block())
+                with gradio.Row(equal_height=True):
+                    __fields.update(create_sampling_block())
+    return __fields
+
+# EVENTS #######################################################################
+
+def generate_password(
+    master_key: str,
+    login_target: str,
+    login_id: str,
+    password_length: int,
+    password_nonce: int,
+    password_level: int,
+    password_alphabet: list,
+) -> str:
+    return gpm.pipeline.process(
+        master_key=master_key,
+        login_target=login_target,
+        login_id=login_id,
+        password_length=password_length,
+        password_nonce=password_nonce,
+        include_lowers=(LOWERS in password_alphabet),
+        include_uppers=(UPPERS in password_alphabet),
+        include_digits=(DIGITS in password_alphabet),
+        include_symbols=(SYMBOLS in password_alphabet),
+        include_spaces=(SPACES in password_alphabet),
+        include_words=(password_level == WORDS),
+        input_vocabulary=[chr(__i) for __i in range(128)],
+        model_context_dim=8,
+        model_embedding_dim=128)
+
+# APP ##########################################################################
+
+def create_app(title: str=TITLE, intro: str=INTRO, style: str=STYLE) -> gradio.Blocks:
+    __fields = {}
+    with gradio.Blocks(theme=gradio.themes.Soft(), title=title, css=style) as __app:
+        # __tokenizer = psaiops.score.similarity.lib.get_tokenizer(name=model, device='cpu')
+        # create the UI
+        __fields.update(create_layout(intro=intro))
+        # init the state
+        __fields.update(create_state())
+        # wire the input fields
+        __fields['process_block'].click(
+            fn=generate_password,
+            inputs=[__fields[__k] for __k in ['key_block', 'target_block', 'identifier_block', 'length_block', 'nonce_block', 'level_block', 'vocabulary_block']],
+            outputs=__fields['password_block'],
+            queue=False,
+            show_progress='full')
+        # gradio application
+        return __app
+
+# MAIN #########################################################################
+
+if __name__ == '__main__':
+    __app = create_app()
+    __app.launch()

requirements.txt

@@ -0,0 +1,2 @@
+gpmanager>=0.4
+gradio>=5.0