core: refacto

app.py

@@ -1,9 +1,6 @@
-import json
-import pandas as pd
 import gradio as gr
 
-from src.chat import respond
-from src.production.flow import play_fn, stop_fn, reset_fn
+from src.ui import sidebar_ui, dashboard_ui
 
 custom_theme = gr.themes.Base(
     primary_hue="blue",
@@ -14,7 +11,7 @@ custom_theme = gr.themes.Base(
 
 with gr.Blocks(theme=custom_theme) as demo:
 
-        # CHAT_INTERFACE
+        # HEADER
         gr.Markdown("# AI Industrial Efficiency Helper ⚡️")
         gr.Markdown("### *AI for efficiency in Industries & Services*")
         gr.Markdown(
@@ -23,65 +20,11 @@ with gr.Blocks(theme=custom_theme) as demo:
             You can interact with the chatbot to get insights and assistance on production-related queries.
             """
         )
-        with gr.Sidebar(width=700, visible=True):
-            gr.Markdown("# Ask Agent")
-            gr.HTML("<div style='margin-bottom: 20px;'></div>")
-            gr.Markdown(
-                """
-                Ask questions about production processes, equipment, and workflows.
-                The chatbot will provide insights and assistance based on the current production data.
-                """
-            )
-            gr.HTML("<div style='margin-bottom: 20px;'></div>")
-            gr.Markdown(
-                """
-                1. **Play** - Start the production simulation and generate synthetic data.
-                2. **Ask Questions** - Interact with the chatbot to get insights on production processes.
-                3. **Ask for Help** - Get assistance with any issues or queries related to production.
-
-                Note: you can click on `stop` or `reset` to control the production simulation.
-                """
-            )
-            gr.HTML("<div style='margin-bottom: 40px;'></div>")
-            chatbot = gr.ChatInterface(respond, type='messages')
+        # CHAT INTERFACE
+        sidebar_ui(width=700, visible=True)
 
         # DASHBOARD
-        with gr.Tab("Dashboard"):
-            with gr.Row():
-                    with gr.Blocks():
-                        play = gr.Button("▶️ Play")
-                        stop = gr.Button("⏸️ Pause")
-                        reset = gr.Button("🔄 Reset")
-
-            with gr.Row():
-                with gr.Column(scale=3):
-                    df_outputs = {
-                        "DataFrame 1": pd.DataFrame(),
-                        "DataFrame 2": pd.DataFrame(),
-                        "DataFrame 3": pd.DataFrame(),
-                        "DataFrame 4": pd.DataFrame(),
-                        "DataFrame 5": pd.DataFrame(),
-                    }
-                    json_output = {}
-
-                    df_components = [gr.DataFrame(label=df_name, visible=True) for df_name in df_outputs.keys()]
-                    json_component = gr.JSON(label="Machine JSON", value=json_output, visible=True)
-
-                    play.click(
-                            fn=play_fn,
-                            inputs=None,
-                            outputs=df_components + [json_component]
-                    )
-                    stop.click(
-                        fn=stop_fn,
-                        inputs=None,
-                        outputs=None
-                    )
-                    reset.click(
-                        fn=reset_fn,
-                        inputs=None,
-                        outputs=None
-                    )
+        dashboard_ui()
 
         # DESCRIPTION
         with gr.Tab("Description"):

src/production/processing.py

@@ -1,5 +1,4 @@
 import json
-import numpy as np
 
 from .metrics.tools import get_tools_metrics
 from .metrics.machine import get_machine_metrics

src/ui/__init__.py

@@ -0,0 +1,2 @@
+from src.ui.sidebar import sidebar_ui
+from src.ui.dashboard import dashboard_ui

src/ui/dashboard.py

@@ -0,0 +1,42 @@
+import gradio as gr
+import pandas as pd
+
+from src.production.flow import play_fn, stop_fn, reset_fn
+
+def dashboard_ui():
+    with gr.Tab("Dashboard"):
+        with gr.Row():
+            with gr.Blocks():
+                play = gr.Button("▶️ Play")
+                stop = gr.Button("⏸️ Pause")
+                reset = gr.Button("🔄 Reset")
+
+        with gr.Row():
+            with gr.Column(scale=3):
+                json_output = {}
+                json_component = gr.JSON(label="Machine JSON", value=json_output, visible=True)
+
+                df_outputs = {
+                    "DataFrame 1": pd.DataFrame(),
+                    "DataFrame 2": pd.DataFrame(),
+                    "DataFrame 3": pd.DataFrame(),
+                    "DataFrame 4": pd.DataFrame(),
+                    "DataFrame 5": pd.DataFrame(),
+                }
+                df_components = [gr.DataFrame(label=df_name, visible=False) for df_name in df_outputs.keys()]
+
+                play.click(
+                    fn=play_fn,
+                    inputs=None,
+                    outputs=df_components + [json_component]
+                )
+                stop.click(
+                    fn=stop_fn,
+                    inputs=None,
+                    outputs=None
+                )
+                reset.click(
+                    fn=reset_fn,
+                    inputs=None,
+                    outputs=None
+                )

src/ui/sidebar.py

@@ -0,0 +1,24 @@
+import gradio as gr
+
+from src.chat import respond
+
+def sidebar_ui(width=700, visible=True):
+    with gr.Sidebar(width=width, visible=visible):
+        gr.Markdown("# Ask Agent")
+        gr.Markdown(
+            """
+            Ask questions about production processes, equipment, and workflows.
+            The chatbot will provide insights and assistance based on the current production data.
+            """
+        )
+        gr.Markdown(
+            """
+            1. **Play** - Start the production simulation and generate synthetic data.
+            2. **Ask Questions** - Interact with the chatbot to get insights on production processes.
+            3. **Ask for Help** - Get assistance with any issues or queries related to production.
+
+            Note: you can click on `stop` or `reset` to control the production simulation.
+            """
+        )
+        gr.HTML("<div style='margin-bottom: 40px;'></div>")
+        chatbot = gr.ChatInterface(respond, type='messages')