debugged minor UI issues

src/fundus_murag/assistant/gemini_fundus_assistant.py

@@ -109,6 +109,8 @@ def list_available_models(only_flash: bool = False) -> pd.DataFrame:
             "output_token_limit": [],
         }
         for model in models:
+            # N = model.name
+            # data["name"].append(N[7:])
             data["name"].append(model.name)
             data["display_name"].append(model.display_name)
             data["input_token_limit"].append(model.input_token_limit)

src/fundus_murag/assistant/openai_fundus_assistant.py

@@ -31,7 +31,7 @@ class OpenAIFundusAssistant(BaseFundusAssistant, metaclass=SingletonMeta):
     def __init__(self, model_name: str, use_tools: bool = True) -> None:
         super().__init__(model_name, use_tools)
         self._conf = load_config()
-        openai.api_key = self._conf.open_ai_api_key
+        openai.api_key = self._conf.open_ai_project_id
 
         self._model_config: ModelConfig = self._create_model_config(
             model_name, use_tools
@@ -103,7 +103,7 @@ def _create_model_config(self, model_name: str, use_tools: bool) -> ModelConfig:
     @staticmethod
     def list_available_models(only_gpt: bool = False) -> pd.DataFrame:
         conf = load_config()
-        openai.api_key = conf.open_ai_api_key
+        openai.api_key = conf.open_ai_project_id
 
         logger.info("Fetching OpenAI models.")
         try:

src/fundus_murag/assistant/openai_image_analysis_assistant.py

@@ -36,7 +36,7 @@ class ModelConfig(TypedDict):
 class OpenAIImageAnalysisAssistant(BaseImageAnalysisAssistant, metaclass=SingletonMeta):
     def __init__(self, model_name: str | None = None):
         conf = load_config()
-        openai.api_key = conf.open_ai_api_key
+        openai.api_key = conf.open_ai_project_id
         self._vdb = VectorDB()
 
         # defult unless we have another model

src/fundus_murag/ui/state.py

@@ -1,7 +1,7 @@
 import mesop as me
 import pandas as pd
 
-from fundus_murag.ui.utils import get_assistant_instance
+from fundus_murag.ui.utils import get_assistant_instance, merge_models
 
 
 @me.stateclass
@@ -14,7 +14,7 @@ class AppState:
 
     available_models: pd.DataFrame
     selected_model_display_name: str = "Gemini 1.5 Flash 002"
-    selected_model: str = "gemini-1.5-flash-002"
+    selected_model: str = "models/gemini-1.5-flash-002"
 
     current_user_input: str
     current_enlarged_record_murag_id: str
@@ -27,19 +27,20 @@ class ModelPickerDialogState:
 
 
 def reset_app_state(state: AppState):
+    state.available_models = pd.DataFrame()
+    state.available_models = merge_models()
+
+    state.selected_model_display_name = "Gemini 1.5 Pro 002"
+    state.selected_model = "models/gemini-1.5-pro-002"
+
     assistant = get_assistant_instance(state.selected_model, state.available_models)
     assistant.reset_chat_session()
 
     state.is_model_picker_dialog_open = False
     state.is_record_dialog_open = False
-
     state.finished_booting = False
     state.current_boot_step = ""
 
-    state.available_models = pd.DataFrame()
-    state.selected_model_display_name = "Gemini 1.5 Pro 002"
-    state.selected_model = "gemini-1.5-pro-002"
-
     state.current_enlarged_record_murag_id = ""
     state.current_user_input = ""