Update app.py

app.py

@@ -126,15 +126,15 @@ def process_sms():
         
         
         logs += ('### ' + min_sec(start) + f' Ответ {a+1}: ')
-        prompt_tokens_a = answer1[a][0].response_metadata['token_usage']['prompt_tokens']
-        completion_tokens_a = answer1[a][0].response_metadata['token_usage']['completion_tokens']
-        total_tokens_a = answer1[a][0].response_metadata['token_usage']['total_tokens']
+        prompt_tokens_a = answer1[a].response_metadata['token_usage']['prompt_tokens']
+        completion_tokens_a = answer1[a].response_metadata['token_usage']['completion_tokens']
+        total_tokens_a = answer1[a].response_metadata['token_usage']['total_tokens']
         logs += f'Токенов промпта: {prompt_tokens_a}'
         logs += f'Токенов сгенерировано: {completion_tokens_a}'
         logs += f'Токенов всего: {total_tokens_a}'
-        prompt_tokens += answer1[a][0].response_metadata['token_usage']['prompt_tokens']
-        completion_tokens += answer1[a][0].response_metadata['token_usage']['completion_tokens']
-        total_tokens += answer1[a][0].response_metadata['token_usage']['total_tokens']
+        prompt_tokens += answer1[a].response_metadata['token_usage']['prompt_tokens']
+        completion_tokens += answer1[a].response_metadata['token_usage']['completion_tokens']
+        total_tokens += answer1[a].response_metadata['token_usage']['total_tokens']
         logs += (f'Сгенерированное сообщение: {answer1[a].content}')
     yield logs