Update app.py

app.py

@@ -337,17 +337,17 @@ def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0,
 
 ВАЖНО: Строго соблюдай формат ответа. Каждый тег должен быть на новой строке:
 
-[THINKING_STEP]
+[THINKING_STEP
 Текст первого шага...
-[/THINKING_STEP]
+[/THINKING_STEP
 
-[THINKING_STEP]
+[THINKING_STEP
 Текст второго шага...
-[/THINKING_STEP]
+[/THINKING_STEP
 
-[FINAL_ANSWER]
+[FINAL_ANSWER
 Итоговый ответ...
-[/FINAL_ANSWER]
+[/FINAL_ANSWER
 
 Вопрос: {message}
 """
@@ -367,8 +367,7 @@ def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0,
                     response = g4f.ChatCompletion.create(
                         model="gpt-4o",
                         messages=[{"role": "system", "content": thinking_prompt}],
-                        stream=True,
-                        # timeout=30
+                        stream=True
                     )
                     
                     # Флаг для отслеживания получения первого чанка
@@ -396,42 +395,42 @@ def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0,
                             )
                             # ... существующий код ...
                             
-                            # Проверяем начало блоков
                             if "[THINKING_STEP]" in current_block:
                                 in_thinking_step = True
                                 current_block = current_block.replace("[THINKING_STEP]", "")
-                            
                             if "[THINKING_STEP" in current_block:
                                 in_thinking_step = True
                                 current_block = current_block.replace("[THINKING_STEP", "")
-                            
+                                
                             if "[FINAL_ANSWER]" in current_block:
                                 in_final_answer = True
                                 current_block = current_block.replace("[FINAL_ANSWER]", "")
-                            
-                            # Проверяем завершение блоков
+                            if "[FINAL_ANSWER" in current_block:
+                                in_final_answer = True
+                                
+                                current_block = current_block.replace("[FINAL_ANSWER", "")
+                                
+                                # Проверяем завершение блоков
                             if "[/THINKING_STEP]" in current_block and in_thinking_step:
                                 block_content = current_block[:current_block.find("[/THINKING_STEP]")]
                                 formatted_block = f'<div class="thinking-step-block">{block_content}</div>'
                                 partial_message += formatted_block
                                 current_block = current_block[current_block.find("[/THINKING_STEP]") + len("[/THINKING_STEP]"):]
                                 in_thinking_step = False
-                            
                             if "[/THINKING_STEP" in current_block and in_thinking_step:
                                 block_content = current_block[:current_block.find("[/THINKING_STEP")]
                                 formatted_block = f'<div class="thinking-step-block">{block_content}</div>'
                                 partial_message += formatted_block
                                 current_block = current_block[current_block.find("[/THINKING_STEP") + len("[/THINKING_STEP"):]
                                 in_thinking_step = False
-                            
-                            if "[/FINAL_ANSWER]" in current_block and in_final_answer:
+                                
+                            elif "[/FINAL_ANSWER]" in current_block and in_final_answer:
                                 block_content = current_block[:current_block.find("[/FINAL_ANSWER]")]
                                 formatted_block = f'<div class="final-answer-block">{block_content}</div>'
                                 partial_message += formatted_block
-                                current_block = current_block[current_block.find("[/FINAL_ANSWER]") + len("[/FINAL_ANSWER]"):]
+                                current_block = current_block[current_block.find("[/FINAL_ANSWER") + len("[/FINAL_ANSWER]"):]
                                 in_final_answer = False
-                            
-                            if "[/FINAL_ANSWER" in current_block and in_final_answer:
+                            elif "[/FINAL_ANSWER" in current_block and in_final_answer:
                                 block_content = current_block[:current_block.find("[/FINAL_ANSWER")]
                                 formatted_block = f'<div class="final-answer-block">{block_content}</div>'
                                 partial_message += formatted_block