Update app.py

app.py

@@ -21,8 +21,8 @@ if check_password():
     base_prompt = st.sidebar.text_area("Промпт для резюмирования", value=initial_base_prompt)
     max_tokens_summary = st.sidebar.number_input("Максимальное количество токенов при резюмировании", min_value=1, value=1024)
 
-    #enable_summarization = st.sidebar.checkbox("Добавить обработку транскрибации", value=False)
-    #processing_prompt = st.sidebar.text_area("Промпт для обработки транскрибации", value=initial_processing_prompt)
+    enable_processing = st.sidebar.checkbox("Добавить обработку транскрибации", value=False)
+    processing_prompt = st.sidebar.text_area("Промпт для обработки транскрибации", value=initial_processing_prompt)
 
     ACCESS_TOKEN = st.secrets["HF_TOKEN"]
 
@@ -78,18 +78,18 @@ if check_password():
         if (llm == 'GigaChat'):
             access_token = get_access_token()
     
-        #if (enable_summarization):
-            #with st.spinner('Обрабатываем транскрибацию...'):
+        if (enable_processing):
+            with st.spinner('Обрабатываем транскрибацию...'):
 
-                #if (llm == 'GigaChat'):
-                    #number_of_tokens = get_number_of_tokens(transcript, access_token)
-                    #print('Количество токенов в транскрибации: ' + str(number_of_tokens))
-                    #transcript = get_completion_from_gigachat(processing_prompt + transcript, number_of_tokens + 1000, access_token)
-                #elif (llm == 'Chat GPT'):
-                    #transcript = get_completion_from_openai(processing_prompt + transcript)
+                if (llm == 'GigaChat'):
+                    number_of_tokens = get_number_of_tokens(transcript, access_token)
+                    print('Количество токенов в транскрибации: ' + str(number_of_tokens))
+                    transcript = get_completion_from_gigachat(processing_prompt + transcript, number_of_tokens + 1000, access_token)
+                elif (llm == 'Chat GPT'):
+                    transcript = get_completion_from_openai(processing_prompt + transcript)
                 
-                #st.write("Результат обработки:")
-                #st.text(transcript)
+                st.write("Результат обработки:")
+                st.text(transcript)