Change function saving

utils.py

@@ -55,10 +55,7 @@ def get_answer(chatbot: list[tuple[str, str]], message: str, model: str) -> tupl
     :return: chatbot answer
     """
     # Setup which function will be called (depends on the model)
-    if COMPANIES[model]['real name'] == 'Gemini':
-        call_model = _call_google
-    else:
-        call_model = _call_openai
+    call_model = COMPANIES[model]['calling']
 
     # Get standalone question
     standalone_question = _get_standalone_question(chatbot, message, call_model)
@@ -226,7 +223,7 @@ PROMPT_GENERAL = download_prompt('general')
 
 # Constants used in the app
 COMPANIES = {
-    'Model G': {'company': 'Google', 'real name': 'Gemini'},
-    'Model C': {'company': 'OpenAI', 'real name': 'ChatGPT 4'},
+    'Model G': {'calling': _call_google, 'real name': 'Gemini'},
+    'Model C': {'calling': _call_openai, 'real name': 'ChatGPT 4'},
 }
 MODELS = list(COMPANIES.keys())