Minor fixes + remove generate buttons for now

dialogs/prompt_variables.py

@@ -17,17 +17,6 @@ def prompt_variables():
     if not state.prompt_variables:
       me.text("No variables defined in prompt.", style=me.Style(width=DIALOG_INPUT_WIDTH))
     else:
-      with me.box(
-        style=me.Style(display="flex", justify_content="end", margin=me.Margin(bottom=15))
-      ):
-        mex.button(
-          "Generate",
-          on_click=on_click_generate_variables,
-          style=me.Style(
-            background=me.theme_var("secondary-container"),
-            color=me.theme_var("on-secondary-container"),
-          ),
-        )
       variable_names = set(parse_variables(state.prompt))
       with me.box(style=me.Style(display="flex", flex_direction="column")):
         for name, value in state.prompt_variables.items():

llm.py

@@ -1,3 +1,4 @@
+from collections.abc import Collection
 import json
 import os
 
@@ -60,7 +61,7 @@ def generate_prompt(task_description: str, model_name: str, temperature: float)
 
 
 def generate_variables(
-  prompt: str, variable_names: list[str], model_name: str, temperature: float
+  prompt: str, variable_names: Collection[str], model_name: str, temperature: float
 ) -> dict[str, str]:
   model = _make_model(model_name, temperature=temperature)
   output = (

main.py

@@ -94,10 +94,13 @@ def app():
     if state.mode == "Prompt":
       # Render prompt creation page
       with me.box(
-        style=me.Style(padding=me.Padding(left=15, top=15, bottom=15), overflow_y="scroll")
+        style=me.Style(padding=me.Padding(left=15, top=15, bottom=15, right=2), overflow_y="scroll")
       ):
         with me.accordion():
-          with me.expansion_panel(title="System Instructions", style=me.Style(background=me.theme_var('surface-container-lowest'))):
+          with me.expansion_panel(
+            title="System Instructions",
+            style=me.Style(background=me.theme_var("surface-container-lowest")),
+          ):
             me.native_textarea(
               autosize=True,
               min_rows=2,
@@ -108,7 +111,11 @@ def app():
               key="system_instructions",
             )
 
-          with me.expansion_panel(title="Prompt", expanded=True, style=me.Style(background=me.theme_var('surface-container-lowest'))):
+          with me.expansion_panel(
+            title="Prompt",
+            expanded=True,
+            style=me.Style(background=me.theme_var("surface-container-lowest")),
+          ):
             me.native_textarea(
               autosize=True,
               min_rows=2,
@@ -120,7 +127,12 @@ def app():
             )
 
         with me.box(
-          style=me.Style(align_items="center", display="flex", justify_content="space-between", margin=me.Margin(top=15))
+          style=me.Style(
+            align_items="center",
+            display="flex",
+            justify_content="space-between",
+            margin=me.Margin(top=15),
+          )
         ):
           with me.content_button(
             type="flat",
@@ -130,25 +142,19 @@ def app():
           ):
             with me.tooltip(message="Run prompt"):
               me.icon("play_arrow")
-          mex.button(
-            "Generate prompt",
-            disabled=bool(state.prompt),
-            style=me.Style(
-              background=me.theme_var("secondary-container"),
-              color=me.theme_var("on-secondary-container"),
-            ),
-            on_click=handlers.on_open_dialog,
-            key="dialog_show_generate_prompt",
-          )
 
       with me.box(style=me.Style(padding=me.Padding.all(15), overflow_y="scroll")):
         if state.response:
-          with me.card(appearance="raised", style=me.Style(background=me.theme_var('surface-container-lowest'))):
+          with me.card(
+            appearance="raised", style=me.Style(background=me.theme_var("surface-container-lowest"))
+          ):
             me.card_header(title="Response")
             with me.card_content():
               mex.markdown(state.response, has_copy_to_clipboard=True)
         else:
-          with me.card(appearance="raised", style=me.Style(background=me.theme_var('surface-container-lowest'))):
+          with me.card(
+            appearance="raised", style=me.Style(background=me.theme_var("surface-container-lowest"))
+          ):
             me.card_header(title="Prompt Tuner Instructions")
             with me.card_content():
               mex.markdown(_INSTRUCTIONS, has_copy_to_clipboard=True)

requirements.txt

@@ -1,3 +1,3 @@
 gunicorn
-mesop==0.12.9
+mesop==0.13.0
 google-generativeai