Update app.py

app.py

@@ -106,9 +106,12 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
         thinking_complete = True
 
         for chunk in response:
+            print("chunk start")
             parts = chunk.candidates[0].content.parts
             current_chunk = parts[0].text
 
+            print(f"\n=========\nparts len: {len(parts)}\n\nparts: {parts}\n\ncurrent chunk: {current_chunk}\n=========\n")
+            
             if len(parts) == 2 and not thinking_complete:
                 # Complete thought and start response
                 thought_buffer += current_chunk
@@ -202,7 +205,15 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
             scale=4
         )
 
-        clear_button = gr.Button("Clear Chat", scale=1)
+        with gr.Column(scale=1):
+            submit_button = gr.Button("Submit", scale=1)
+            clear_button = gr.Button("Clear Chat", scale=1)
+
+    with gr.Row(equal_height=True):
+        test_button = gr.Button("test", scale=1)
+        test1_button = gr.Button("test1", scale=1)
+        test2_button = gr.Button("test2", scale=1)
+        test3_button = gr.Button("test3", scale=1)
 
     # Add example prompts - removed file upload examples. Kept text focused examples.
     example_prompts = [
@@ -220,7 +231,28 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
         examples_per_page=5 # Adjust as needed
     )
 
-
+# Created by gemini-2.5-pro-exp-03-25
+#def process_message(msg):
+#    """Обрабатывает сообщение пользователя: сохраняет, отображает и генерирует ответ."""
+#    msg_store_val, _, _ = lambda msg: (msg, msg, "")(msg) # Store message and clear input (inline lambda)
+#    input_box_val, chatbot_val = user_message(msg_store_val, chatbot) # Add user message to chat
+#    chatbot_val_final = stream_gemini_response(msg_store_val, chatbot_val) # Generate and stream response
+#    return msg_store_val, input_box_val, chatbot_val_final
+#
+#input_box.submit(
+#    process_message,
+#    inputs=[input_box],
+#    outputs=[msg_store, input_box, chatbot],  # Исправлены outputs, чтобы включать chatbot
+#    queue=False
+#)
+
+#submit_button.click(
+#    process_message,
+#    inputs=[input_box],
+#    outputs=[msg_store, input_box, chatbot],  # Исправлены outputs, чтобы включать chatbot
+#    queue=False
+#)
+    
     # Set up event handlers
     msg_store = gr.State("")  # Store for preserving user message
 
@@ -240,6 +272,22 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
         outputs=chatbot
     )
 
+    submit_button.click(
+        lambda msg: (msg, msg, ""),  # Store message and clear input
+        inputs=[input_box],
+        outputs=[msg_store, input_box, input_box],
+        queue=False
+    ).then(
+        user_message,  # Add user message to chat
+        inputs=[msg_store, chatbot],
+        outputs=[input_box, chatbot],
+        queue=False
+    ).then(
+        stream_gemini_response,  # Generate and stream response
+        inputs=[msg_store, chatbot],
+        outputs=chatbot
+    )
+
     clear_button.click(
         lambda: ([], "", ""),
         outputs=[chatbot, input_box, msg_store],