v.0.0.2 (#1)

- Summary (69f6759227a857c3a1ddc012f3f7b896fb7d7ad3)


Co-authored-by: Dmitry Ryumin <DmitryRyumin@users.noreply.huggingface.co>

.gitignore

@@ -169,4 +169,5 @@ dmypy.json
 
 # Custom
 logs/
-models/
+models/
+*.csv

README.md

@@ -1,6 +1,6 @@
 ---
 title: OCEANAI
-emoji: 🤩🔝🔥
+emoji: 😀🤓😎😉😤
 colorFrom: gray
 colorTo: red
 sdk: gradio

app.css

@@ -15,13 +15,15 @@ div.files-container {
     max-height: 350px;
 }
 
+div.csv-container {}
+
 div.files-container div.file-preview-holder {
     overflow-y: scroll;
     height: 100%;
 }
 
 div.files-container label[data-testid="block-label"] {
-  position: absolute;
+    position: absolute;
 }
 
 .calculate_oceanai {
@@ -39,7 +41,7 @@ div.files-container label[data-testid="block-label"] {
     transition: all 0.3s ease;
 }
 
-.clear_oceanai {
+.examples_oceanai, .clear_oceanai {
     display: inline-block;
     padding: 10px 20px;
     font-size: 16px;
@@ -51,7 +53,9 @@ div.files-container label[data-testid="block-label"] {
     transition: all 0.3s ease;
 }
 
-.calculate_oceanai[disabled], .clear_oceanai[disabled] {
+.calculate_oceanai[disabled],
+.examples_oceanai[disabled],
+.clear_oceanai[disabled] {
     cursor: not-allowed;
     opacity: 0.6;
 }
@@ -62,6 +66,8 @@ div.files-container label[data-testid="block-label"] {
     color: var(--button-primary-text-color-hover);
 }
 
-.calculate_oceanai:active:not([disabled]), .clear_oceanai:active:not([disabled]) {
+.calculate_oceanai:active:not([disabled]),
+.examples_oceanai:active:not([disabled]),
+.clear_oceanai:active:not([disabled]) {
     transform: scale(0.98);
 }

app/components.py

@@ -87,8 +87,13 @@ def button(
     value: str = "",
     interactive: bool = True,
     scale: int = 3,
+    visible: bool = True,
     elem_classes: Optional[str] = None,
 ) -> gr.Button:
     return gr.Button(
-        value=value, interactive=interactive, scale=scale, elem_classes=elem_classes
+        value=value,
+        interactive=interactive,
+        scale=scale,
+        visible=visible,
+        elem_classes=elem_classes,
     )

app/description.py

@@ -9,7 +9,7 @@ License: MIT License
 from app.config import config_data
 
 DESCRIPTION = f"""\
-# OCEANAI
+<h1><a href="https://github.com/aimclub/OCEANAI" target="_blank">OCEAN-AI</a> is an open-source library to Big Five personality traits assessment and automate HR-processes.</h1>
 
 <div class="app-flex-container">
     <img src="https://img.shields.io/badge/version-v{config_data.AppSettings_APP_VERSION}-rc0" alt="Version">

app/event_handlers/calculate_pt_scores_blocks.py

@@ -10,7 +10,7 @@ import gradio as gr
 # Importing necessary components for the Gradio app
 from app.oceanai_init import b5
 from app.config import config_data
-from app.components import html_message, dataframe
+from app.components import html_message, dataframe, files_create_ui
 
 
 def event_handler_calculate_pt_scores_blocks(files, evt_data: gr.EventData):
@@ -21,6 +21,8 @@ def event_handler_calculate_pt_scores_blocks(files, evt_data: gr.EventData):
         paths=files, url_accuracy="", accuracy=False, lang="en", out=out
     )
 
+    b5.df_files_.to_csv(config_data.Filenames_PT_SCORES)
+
     return (
         html_message(config_data.InformationMessages_NOTI_VIDEOS, False, False),
         dataframe(
@@ -28,4 +30,14 @@ def event_handler_calculate_pt_scores_blocks(files, evt_data: gr.EventData):
             values=b5.df_files_.values.tolist(),
             visible=True,
         ),
+        files_create_ui(
+            config_data.Filenames_PT_SCORES,
+            "single",
+            [".csv"],
+            config_data.OtherMessages_EXPORT_PT_SCORES,
+            True,
+            False,
+            True,
+            "csv-container",
+        ),
     )

app/event_handlers/clear_blocks.py

@@ -25,8 +25,19 @@ def event_handler_clear_blocks():
             config_data.OtherMessages_CALCULATE_PT_SCORES,
             False,
             3,
+            True,
             "calculate_oceanai",
         ),
-        button(config_data.OtherMessages_CLEAR_APP, False, 1, "clear_oceanai"),
+        button(config_data.OtherMessages_CLEAR_APP, False, 1, True, "clear_oceanai"),
         dataframe(visible=False),
+        files_create_ui(
+            None,
+            "single",
+            [".csv"],
+            config_data.OtherMessages_EXPORT_PT_SCORES,
+            True,
+            False,
+            False,
+            "csv-container",
+        ),
     )

app/event_handlers/event_handlers.py

@@ -12,6 +12,7 @@ from app.event_handlers.files import (
     event_handler_files,
     event_handler_files_select,
 )
+from app.event_handlers.examples_blocks import event_handler_examples_blocks
 from app.event_handlers.clear_blocks import event_handler_clear_blocks
 from app.event_handlers.calculate_pt_scores_blocks import (
     event_handler_calculate_pt_scores_blocks,
@@ -19,7 +20,14 @@ from app.event_handlers.calculate_pt_scores_blocks import (
 
 
 def setup_app_event_handlers(
-    notifications, files, video, calculate_pt_scores, clear_app, pt_scores
+    notifications,
+    files,
+    video,
+    examples,
+    calculate_pt_scores,
+    clear_app,
+    pt_scores,
+    csv_pt_scores,
 ):
     # Events
     files.change(
@@ -38,7 +46,15 @@ def setup_app_event_handlers(
         triggers=[calculate_pt_scores.click],
         fn=event_handler_calculate_pt_scores_blocks,
         inputs=[files],
-        outputs=[notifications, pt_scores],
+        outputs=[notifications, pt_scores, csv_pt_scores],
+        queue=True,
+    )
+    examples.click(
+        fn=event_handler_examples_blocks,
+        inputs=[],
+        outputs=[
+            files,
+        ],
         queue=True,
     )
     clear_app.click(
@@ -51,6 +67,7 @@ def setup_app_event_handlers(
             calculate_pt_scores,
             clear_app,
             pt_scores,
+            csv_pt_scores,
         ],
         queue=True,
     )

app/event_handlers/examples_blocks.py

@@ -0,0 +1,16 @@
+"""
+File: examples_blocks.py
+Author: Elena Ryumina and Dmitry Ryumin
+Description: Event handler for the addition of examples to the Gradio app.
+License: MIT License
+"""
+
+# Importing necessary components for the Gradio app
+
+
+def event_handler_examples_blocks():
+    return [
+        "videos/video1.mp4",
+        "videos/video2.mp4",
+        "videos/video3.mp4",
+    ]

app/event_handlers/files.py

@@ -22,9 +22,12 @@ def event_handler_files(files):
                 config_data.OtherMessages_CALCULATE_PT_SCORES,
                 False,
                 3,
+                True,
                 "calculate_oceanai",
             ),
-            button(config_data.OtherMessages_CLEAR_APP, False, 1, "clear_oceanai"),
+            button(
+                config_data.OtherMessages_CLEAR_APP, False, 1, True, "clear_oceanai"
+            ),
         )
 
     return (
@@ -34,9 +37,10 @@ def event_handler_files(files):
             config_data.OtherMessages_CALCULATE_PT_SCORES,
             True,
             3,
+            True,
             "calculate_oceanai",
         ),
-        button(config_data.OtherMessages_CLEAR_APP, True, 1, "clear_oceanai"),
+        button(config_data.OtherMessages_CLEAR_APP, True, 1, True, "clear_oceanai"),
     )
 
 

app/tabs.py

@@ -29,21 +29,45 @@ def app_tab():
         video = video_create_ui()
 
     with gr.Row():
+        examples = button(
+            config_data.OtherMessages_EXAMPLES_APP, True, 1, True, "examples_oceanai"
+        )
         calculate_pt_scores = button(
             config_data.OtherMessages_CALCULATE_PT_SCORES,
             False,
             3,
+            True,
             "calculate_oceanai",
         )
         clear_app = button(
-            config_data.OtherMessages_CLEAR_APP, False, 1, "clear_oceanai"
+            config_data.OtherMessages_CLEAR_APP, False, 1, True, "clear_oceanai"
         )
 
     notifications = html_message(config_data.InformationMessages_NOTI_VIDEOS, False)
 
     pt_scores = dataframe(visible=False)
 
-    return (notifications, files, video, calculate_pt_scores, clear_app, pt_scores)
+    csv_pt_scores = files_create_ui(
+        None,
+        "single",
+        [".csv"],
+        config_data.OtherMessages_EXPORT_PT_SCORES,
+        True,
+        False,
+        False,
+        "csv-container",
+    )
+
+    return (
+        notifications,
+        files,
+        video,
+        examples,
+        calculate_pt_scores,
+        clear_app,
+        pt_scores,
+        csv_pt_scores,
+    )
 
 
 def about_authors_tab():

config.toml

@@ -1,5 +1,5 @@
 [AppSettings]
-APP_VERSION = "0.1.0"
+APP_VERSION = "0.2.0"
 CSS_PATH = "app.css"
 
 [InformationMessages]
@@ -8,9 +8,13 @@ NOTI_VIDEOS = "Select the video(s)"
 [OtherMessages]
 CALCULATE_PT_SCORES = "Calculation of Big Five personality traits scores"
 CLEAR_APP = "Clear"
+EXAMPLES_APP = "Examples"
+EXPORT_PT_SCORES = "Export Big Five personality traits to a CSV file"
 NOTI_CALCULATE = "You can calculate Big Five personality traits scores"
 
 [TabCreators]
 "App" = "app_tab"
 "About the Authors" = "about_authors_tab"
 
+[Filenames]
+PT_SCORES = "personality_traits_scores.csv"