fixed llm_sanitize

chains/exercises/runner.py

@@ -3,6 +3,8 @@ import asyncio
 from typing import AsyncGenerator
 from config.llm_config import llms
 from config.chain_configs import chain_configs
+from config.templates import template_sanitize_fluster
+
 
 async def run_fluster(
     user_input_text: str,

config/chain_configs.py

@@ -75,6 +75,6 @@ chain_configs = {
         "template_refine_fluster": template_refine_fluster,
         "llm_refine": llms["o1 (high reasoning_effort)"],
         "template_sanitize": template_sanitize_fluster,
-        "llm_sanitize": "GPT-4o-mini (zero temp)",
+        "llm_sanitize": llms["GPT-4o (zero temp)"],
     },
 }

main.py

@@ -99,7 +99,13 @@ with gr.Blocks() as interface:
              [fluster_box_0, fluster_box_1, fluster_box_2, fluster_box_3],
              ) = build_write_fluster_tab()
 
-            # 3 Empty separators (somehow scale=3 doesn't work)
+            # 6 Empty separators (somehow scale=6 doesn't work)
+            with gr.Tab("", visible=True):
+                pass
+            with gr.Tab("", visible=True):
+                pass
+            with gr.Tab("", visible=True):
+                pass
             with gr.Tab("", visible=True):
                 pass
             with gr.Tab("", visible=True):