Only show openrouter models on huggingface spaces

src/plan/app_text2plan.py

@@ -31,15 +31,18 @@ IS_HUGGINGFACE_SPACES = is_huggingface_spaces()
 @dataclass
 class Config:
     use_uuid_as_run_id: bool
-    enable_open_output_dir_button: bool
+    visible_open_output_dir_button: bool
+    allow_only_openrouter_models: bool
 
 CONFIG_LOCAL = Config(
     use_uuid_as_run_id=False,
-    enable_open_output_dir_button=True
+    visible_open_output_dir_button=True,
+    allow_only_openrouter_models=False,
 )
 CONFIG_HUGGINGFACE_SPACES = Config(
     use_uuid_as_run_id=True,
-    enable_open_output_dir_button=False
+    visible_open_output_dir_button=False,
+    allow_only_openrouter_models=True,
 )
 
 CONFIG = CONFIG_HUGGINGFACE_SPACES if IS_HUGGINGFACE_SPACES else CONFIG_LOCAL
@@ -70,8 +73,11 @@ gradio_examples = []
 for prompt_item in all_prompts:
     gradio_examples.append([prompt_item.prompt])
 
-available_models = get_available_llms()
-available_model_names = [model for model in available_models]
+all_available_models = get_available_llms()
+if CONFIG.allow_only_openrouter_models:
+    available_model_names = [model for model in all_available_models if model.startswith("openrouter")]
+else:
+    available_model_names = all_available_models
 
 def has_pipeline_complete_file(path_dir: str):
     """
@@ -380,7 +386,7 @@ with gr.Blocks(title="PlanExe") as demo_text2plan:
                     submit_btn = gr.Button("Submit", variant='primary')
                     stop_btn = gr.Button("Stop")
                     retry_btn = gr.Button("Retry")
-                    open_dir_btn = gr.Button("Open Output Dir", visible=CONFIG.enable_open_output_dir_button)
+                    open_dir_btn = gr.Button("Open Output Dir", visible=CONFIG.visible_open_output_dir_button)
 
                 output_markdown = gr.Markdown("Output will appear here...")
                 status_markdown = gr.Markdown("Status messages will appear here...")