Update app.py

app.py

@@ -96,9 +96,10 @@ def process_file(file):
         return None
         
     try:
+        # print()
         # Получаем расширение файла
         file_extension = file.name.lower().split('.')[-1] if hasattr(file, 'name') else ''
-        
+        print(file_extension)
         # Список поддерживаемых форматов изображений
         image_extensions = {'png', 'jpg', 'jpeg', 'gif', 'bmp', 'webp'}
         
@@ -278,7 +279,7 @@ def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0,
                 for link in results:
                     info_text += f"Ссылка: {link}\n"
                     page_text = get_page_text(link)
-                    info_text += f"Текст: {page_text}...\n\n"
+                    info_text += f"Текст: {page_text[:500]}...\n\n"
                 
                 search_response_prompt = f"""Вот информация по запросу "{prompt}":
                 
@@ -297,8 +298,36 @@ def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0,
                 final_response = final_response.replace(f"[SEARCH: {prompt}]", "").strip()
                 history.append({"role": "assistant", "content": final_response})
                 yield history, *[gr.update(visible=False) for _ in range(6)]
-                return
+                
+                if response and isinstance(response, str):
+                    steps = response.split("[THINKING_STEP]")
+                    final_answer = ""
+
+                    for i, step in enumerate(steps):
+                        if "[FINAL_ANSWER]" in step:
+                            final_answer = step.split("[FINAL_ANSWER]")[1].strip()
+                            step = step.split("[FINAL_ANSWER]")[0]
+
+                        step_html = f"""
+                        <div class="thinking-step-block">
+                            <span class="thinking-step">Шаг {i + 1}: {step.strip()}</span>
+                        </div>
+                        """
+                        history[-1]["content"] += step_html
+
+                        yield history, gr.update(visible=False)
+
+                    # Добавляем финальный блок с заключением
+                    final_html = f"""
+                    <div class="final-answer-block">
+                        <span class="thinking-conclusion">Ответ: {final_answer}</span>
+                    </div>
+                    """
+                    history[-1]["content"] += final_html
+                    yield history, gr.update(visible=True)
+                    return
 
+    
         history[-1]["content"] = partial_message.strip()
         yield history, *[gr.update(visible=True) for _ in range(6)]
 
@@ -344,7 +373,12 @@ def chat_response(message, history, model_name, direct_mode, thinking_depth=1.0,
 
     partial_message = ""
     code_block = None
-
+    # response = g4f.ChatCompletion.create(
+    #     model=AVAILABLE_MODELS.get(model_name, "O3"),
+    #     messages=messages,
+    #     stream=False,
+    #     provider="PollinationsAI"
+    # )
     
     if response and isinstance(response, str):
         partial_message += response
@@ -803,12 +837,53 @@ def create_interface():
         from {
             opacity: 0;
             transform: translateX(-10px);
-        }
+        }                  
         to {
             opacity: 1;
             transform: translateX(0);
         }
     }
+    .thinking-animation {
+    position: relative;
+    width: 100%;
+    min-height: 100px;
+    background: linear-gradient(45deg, #1a1a1a, #2d2d2d);
+    border-radius: 15px;
+    overflow: hidden;
+    margin: 10px 0;
+    padding: 20px;
+    border-left: 4px solid #4a90e2;
+    transition: all 0.3s ease;
+}
+
+.thinking-step {
+    color: #b0b0b0;
+    font-style: italic;
+    margin-bottom: 10px;
+    padding-left: 20px;
+    border-left: 2px solid #4a90e2;
+    opacity: 0;
+    transform: translateY(10px);
+    animation: fadeInStep 0.5s ease-in-out forwards;
+}
+
+.thinking-conclusion {
+    color: white;
+    font-weight: bold;
+    margin-top: 15px;
+    padding: 10px;
+    background: rgba(74, 144, 226, 0.1);
+    border-radius: 8px;
+    border-left: 4px solid #4a90e2;
+}
+
+@keyframes fadeInStep {
+    to {
+        opacity: 1;
+        transform: translateY(0);
+    }
+}
+
     """
 
     # Добавте этот скрипт в create_interface()
@@ -849,7 +924,7 @@ def create_interface():
                         scale=1,
                         elem_classes="submit-btn"
                     )
-                    clear = gr.Button("Очистить", scale=1)
+                    # clear = gr.Button("Очистить", scale=1)
             
             with gr.Column(scale=1, visible=True) as sidebar:
                 model = gr.Dropdown(
@@ -917,11 +992,10 @@ def create_interface():
         )
         
 
-        
-        clear.click(
-            fn=lambda: (None, "", "", False, ""),
-            outputs=[chatbot, current_code]
-        )
+        # clear.click(
+        #     fn=lambda: (None, ""),
+        #     outputs=[chatbot, current_code]
+        # )
 
         # Добавляем обработчик изменения модели
         def update_thinking_depth_visibility(model_name):