Update app.py

app.py

@@ -1,162 +1,148 @@
-import faicons as fa
-import plotly.express as px
-
-# Load data and compute static values
-from shared import app_dir, tips
-from shinywidgets import render_plotly
-
-from shiny import reactive, render
-from shiny.express import input, ui
-
-bill_rng = (min(tips.total_bill), max(tips.total_bill))
-
-# Add page title and sidebar
-ui.page_opts(title="Restaurant tipping", fillable=True)
-
-with ui.sidebar(open="desktop"):
-    ui.input_slider(
-        "total_bill",
-        "Bill amount",
-        min=bill_rng[0],
-        max=bill_rng[1],
-        value=bill_rng,
-        pre="$",
-    )
-    ui.input_checkbox_group(
-        "time",
-        "Food service",
-        ["Lunch", "Dinner"],
-        selected=["Lunch", "Dinner"],
-        inline=True,
-    )
-    ui.input_action_button("reset", "Reset filter")
-
-# Add main content
-ICONS = {
-    "user": fa.icon_svg("user", "regular"),
-    "wallet": fa.icon_svg("wallet"),
-    "currency-dollar": fa.icon_svg("dollar-sign"),
-    "ellipsis": fa.icon_svg("ellipsis"),
-}
-
-with ui.layout_columns(fill=False):
-    with ui.value_box(showcase=ICONS["user"]):
-        "Total tippers"
-
-        @render.express
-        def total_tippers():
-            tips_data().shape[0]
-
-    with ui.value_box(showcase=ICONS["wallet"]):
-        "Average tip"
-
-        @render.express
-        def average_tip():
-            d = tips_data()
-            if d.shape[0] > 0:
-                perc = d.tip / d.total_bill
-                f"{perc.mean():.1%}"
-
-    with ui.value_box(showcase=ICONS["currency-dollar"]):
-        "Average bill"
-
-        @render.express
-        def average_bill():
-            d = tips_data()
-            if d.shape[0] > 0:
-                bill = d.total_bill.mean()
-                f"${bill:.2f}"
-
-
-with ui.layout_columns(col_widths=[6, 6, 12]):
-    with ui.card(full_screen=True):
-        ui.card_header("Tips data")
-
-        @render.data_frame
-        def table():
-            return render.DataGrid(tips_data())
-
-    with ui.card(full_screen=True):
-        with ui.card_header(class_="d-flex justify-content-between align-items-center"):
-            "Total bill vs tip"
-            with ui.popover(title="Add a color variable", placement="top"):
-                ICONS["ellipsis"]
-                ui.input_radio_buttons(
-                    "scatter_color",
-                    None,
-                    ["none", "sex", "smoker", "day", "time"],
-                    inline=True,
+from pathlib import Path
+
+import uvicorn
+from faicons import icon_svg as icon
+from fire import Fire
+from shiny import App, reactive, render, ui
+
+from quickmt import Translator
+from quickmt.hub import hf_download, hf_list
+
+t = None
+
+port: int = 7860,
+host: str = "0.0.0.0"
+ui.navbar_options(
+    bg="red",
+)
+app_ui = ui.page_navbar(
+    ui.nav_panel(
+        None,
+        ui.layout_columns(
+            ui.card(
+                ui.h4("Input Text"),
+                ui.input_text_area(
+                    "input_text",
+                    "",
+                    value="",
+                    width="100%",
+                    height="600px",
+                ),
+                ui.input_action_button(
+                    "translate_button", "Translate!", class_="btn-primary"
+                ),
+            ),
+            ui.card(ui.h4("Translation"), ui.output_ui("translate")),
+        ),
+    ),
+    ui.nav_spacer(),
+    ui.nav_control(
+        ui.input_dark_mode(
+            id="darkmode_toggle", mode="dark", style="padding-top: 10px;"
+        ),
+    ),
+    ui.nav_control(
+        ui.a(
+            icon("github", height="30px", width="30px", fill="#17a2b8"),
+            href="https://github.com/quickmt/quickmt",
+            target="_blank",
+            class_="btn btn-link",
+        ),
+    ),
+    sidebar=ui.sidebar(
+        ui.tooltip(
+            ui.input_selectize(
+                "model",
+                "Select model",
+                choices=[i.split("/")[1] for i in hf_list()],
+            ),
+            "QuickMT model to use. quickmt-fr-en will translate from French (fr) to English (en)",
+        ),
+        ui.tooltip(
+            ui.input_text(
+                "model_folder", "Model Folder", value=str(Path(".").absolute())
+            ),
+            "Folder where QuickMT models are (or will be) stored.",
+        ),
+        ui.tooltip(
+            ui.input_slider(
+                "beam_size", "Beam size", min=1, max=8, step=1, value=2
+            ),
+            "Balances speed and quality. 1 for fastest speed, 8 for highest quality, in between for a balance.",
+        ),
+        ui.tooltip(
+            ui.input_numeric(
+                "num_threads", "CPU Threads", min=1, max=16, step=1, value=4
+            ),
+            "Number of CPU threads to use for translation. Does not affect speed when using GPU.",
+        ),
+        ui.tooltip(
+            ui.input_selectize(
+                "compute_device",
+                "Compute Device",
+                choices=["auto", "cpu", "cuda"],
+                selected="cpu",
+            ),
+            "Auto will use the GPU if available, otherwise will use CPU.",
+        ),
+        width="350px",
+    ),
+    title=ui.h2("QuickMT"),
+    window_title="QuickMT",
+    theme=ui.Theme.from_brand(__file__),
+    navbar_options=ui.navbar_options(underline=False, theme="auto"),
+)
+
+def server(input, output, session):
+    @render.ui
+    @reactive.event(input.quickmt_model_download)  # Take a dependency on the button
+    def model_download_output():
+        print(f"Downloading {input.model()} to {input.model_folder()}")
+        hf_download(
+            model_name="quickmt/" + input.model(),
+            output_dir=Path(input.model_folder()) / input.model(),
+        )
+        return "Model downloaded"
+
+    @render.ui
+    @reactive.event(input.translate_button)  # Take a dependency on the button
+    def translate():
+        global t
+        model_path = Path(input.model_folder()) / input.model()
+        try:
+            if t is None or str(input.model()) != str(Path(t.model_path).name):
+                print(f"Loading model {input.model()}")
+                t = Translator(
+                    str(model_path),
+                    device=input.compute_device(),
+                    inter_threads=int(input.num_threads()),
                 )
+            if len(input.input_text()) == 0:
+                return ""
 
-        @render_plotly
-        def scatterplot():
-            color = input.scatter_color()
-            return px.scatter(
-                tips_data(),
-                x="total_bill",
-                y="tip",
-                color=None if color == "none" else color,
-                trendline="lowess",
-            )
-
-    with ui.card(full_screen=True):
-        with ui.card_header(class_="d-flex justify-content-between align-items-center"):
-            "Tip percentages"
-            with ui.popover(title="Add a color variable"):
-                ICONS["ellipsis"]
-                ui.input_radio_buttons(
-                    "tip_perc_y",
-                    "Split by:",
-                    ["sex", "smoker", "day", "time"],
-                    selected="day",
-                    inline=True,
+            return [
+                ui.p(i)
+                for i in t(
+                    input.input_text().splitlines(), beam_size=input.beam_size()
                 )
+            ]
 
-        @render_plotly
-        def tip_perc():
-            from ridgeplot import ridgeplot
-
-            dat = tips_data()
-            dat["percent"] = dat.tip / dat.total_bill
-            yvar = input.tip_perc_y()
-            uvals = dat[yvar].unique()
-
-            samples = [[dat.percent[dat[yvar] == val]] for val in uvals]
-
-            plt = ridgeplot(
-                samples=samples,
-                labels=uvals,
-                bandwidth=0.01,
-                colorscale="viridis",
-                colormode="row-index",
-            )
-
-            plt.update_layout(
-                legend=dict(
-                    orientation="h", yanchor="bottom", y=1.02, xanchor="center", x=0.5
-                )
-            )
-
-            return plt
-
-
-ui.include_css(app_dir / "styles.css")
-
-# --------------------------------------------------------
-# Reactive calculations and effects
-# --------------------------------------------------------
+        except:
+            return [
+                ui.value_box(
+                    title=f"Ensure model is downloaded to {input.model_folder()}",
+                    value="Failed to load model",
+                    showcase=icon("bug"),
+                ),
+                ui.input_action_button(
+                    "quickmt_model_download", "Download Model", class_="btn-primary"
+                ),
+                ui.output_ui("model_download_output"),
+            ]
 
+app = App(app_ui, server)
 
-@reactive.calc
-def tips_data():
-    bill = input.total_bill()
-    idx1 = tips.total_bill.between(bill[0], bill[1])
-    idx2 = tips.time.isin(input.time())
-    return tips[idx1 & idx2]
+if __name__=="__main__":
+    uvicorn.run(app, port=port, host=host)
 
 
-@reactive.effect
-@reactive.event(input.reset)
-def _():
-    ui.update_slider("total_bill", value=bill_rng)
-    ui.update_checkbox_group("time", selected=["Lunch", "Dinner"])