Update app.py

app.py

@@ -19,7 +19,16 @@ if check_password():
 
     min_speakers = st.sidebar.number_input("Минимальное количество спикеров", min_value=1, value=2)
     max_speakers = st.sidebar.number_input("Максимальное количество спикеров", min_value=1, value=2)
-    llm = st.sidebar.selectbox("LLM", ["GigaChat", "Chat GPT"], index=0)
+    llm = st.sidebar.selectbox("Производитель LLM", ["Сбер", "OpenAI"], index=0)
+
+    if llm == "Сбер":
+        options = ["GigaChat-Plus", "GigaChat", "GigaChat-Pro"]
+    elif llm == "OpenAI":
+        options = ["gpt-4o", "gpt-4o-mini", "gpt-4-turbo", "gpt-4", "gpt-3.5-turbo"]
+    else:
+        options = []
+
+    model = st.sidebar.selectbox("Модель", options, index=0)
     base_prompt = st.sidebar.text_area("Промпт для резюмирования", value=initial_base_prompt)
 
     enable_processing = st.sidebar.checkbox("Добавить обработку транскрибации", value=False)
@@ -76,27 +85,27 @@ if check_password():
         st.write("Результат транскрибации:")
         st.text(transcript)
 
-        if (llm == 'GigaChat'):
+        if (llm == 'Сбер'):
             access_token = get_access_token()
     
         if (enable_processing):
             with st.spinner('Обрабатываем транскрибацию...'):
 
-                if (llm == 'GigaChat'):
-                    number_of_tokens = get_number_of_tokens(transcript, access_token)
+                if (llm == 'Сбер'):
+                    number_of_tokens = get_number_of_tokens(transcript, access_token, model)
                     print('Количество токенов в транскрибации: ' + str(number_of_tokens))
-                    transcript = process_transcribation_with_gigachat(processing_prompt, transcript, number_of_tokens + 1000, access_token)
+                    transcript = process_transcribation_with_gigachat(processing_prompt, transcript, number_of_tokens + 1000, access_token, model)
                     print(transcript)
                     
-                elif (llm == 'Chat GPT'):
+                elif (llm == 'OpenAI'):
                     transcript = process_transcribation_with_assistant(processing_prompt, transcript)
                     print(transcript)
     
         with st.spinner('Резюмируем...'):
-            if (llm == 'GigaChat'):
-                summary_answer = get_completion_from_gigachat(base_prompt + transcript, 1024, access_token)
-            elif (llm == 'Chat GPT'):
-                summary_answer = get_completion_from_openai(base_prompt + transcript, 1024)
+            if (llm == 'Сбер'):
+                summary_answer = get_completion_from_gigachat(base_prompt + transcript, 1024, access_token, model)
+            elif (llm == 'OpenAI'):
+                summary_answer = get_completion_from_openai(base_prompt + transcript, 1024, model)
         
             st.write("Результат резюмирования:")
             st.text(summary_answer)