New UI animations!

app.py

@@ -8,6 +8,7 @@ import json
 import time
 import base64
 import tempfile
+import subprocess
 from datetime import datetime, timedelta
 from openai import OpenAI
 
@@ -46,7 +47,12 @@ SESSION_TIMEOUT_SEC = 15 * 60
 def clean_thought_process(text):
     """Removes the <think>...</think> blocks from the text to prevent leaks."""
     if not text: return ""
-    return re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL).strip()
+    # Remove completed blocks
+    text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
+    # Remove pending/unclosed blocks
+    if "<think>" in text:
+        text = text.split("<think>")[0]
+    return text.strip()
 
 def export_script(code):
     if not code.strip(): return None
@@ -67,44 +73,37 @@ def detect_language(filename):
     return "python"
 
 def robust_search(query):
-    if not SEARCH_AVAILABLE: return json.dumps({"error": "Missing duckduckgo-search"})
+    if not SEARCH_AVAILABLE: return "Error: Missing duckduckgo-search"
     try:
         results = []
         with DDGS() as ddgs:
-            # Fetch 5 results
             ddg_gen = ddgs.text(query, max_results=5)
             if ddg_gen:
                 for r in ddg_gen:
-                    results.append({
-                        "title": r.get('title'),
-                        "link": r.get('href'),
-                        "snippet": r.get('body')
-                    })
-        return json.dumps(results, indent=2)
+                    results.append(f"### [{r.get('title')}]({r.get('href')})\n{r.get('body')}\n")
+        
+        if not results: return "⚠️ No results found."
+        return "\n".join(results)
     except Exception as e:
-        return json.dumps({"error": str(e)})
+        return f"❌ Search Error: {str(e)}"
 
 # ==================== TOOL: E2B SANDBOX (POLYGLOT) ====================
 
 def execute_code_e2b(project_state, mode="Code Mode"):
     """
     Polymorphic execution engine with Mode-Gated Internet Access.
-    - Web Mode: Internet ON
-    - Other Modes: Internet OFF (Air-gapped)
     """
     if not E2B_AVAILABLE or not E2B_API_KEY:
         return "❌ Error: E2B API Key missing. Cannot execute code."
 
-    # Gate Internet Access based on Mode
     enable_internet = (mode == "Web Mode")
-    
     logger.info(f"⚡ Executing in E2B Sandbox (Mode: {mode}, Internet: {enable_internet})...")
     
     try:
         with Sandbox.create(allow_internet_access=enable_internet) as sandbox:
             # 1. Write all auxiliary files to the sandbox filesystem
             for filename, content in project_state.items():
-                if filename not in ["main.py", "main.cpp"]: # Write aux files first
+                if filename not in ["main.py", "main.cpp"]: 
                     sandbox.files.write(filename, content)
             
             output_log = ""
@@ -115,7 +114,7 @@ def execute_code_e2b(project_state, mode="Code Mode"):
             if "main.cpp" in project_state:
                 logger.info("⚙️ Detected C++ Project. Compiling via Python wrapper...")
                 
-                # Write C++ source
+                # Write C++ source explicitly ensures it's fresh
                 sandbox.files.write("main.cpp", project_state["main.cpp"])
                 
                 # Trojan Horse: Use Python to manage the compilation process reliably
@@ -140,7 +139,7 @@ else:
 """
                 execution = sandbox.run_code(cpp_runner_script)
                 
-                # FIXED: Properly join list of strings into a single string
+                # Properly join list of strings into a single string
                 if execution.logs.stdout:
                     output_log += "".join(execution.logs.stdout) + "\n"
                 if execution.logs.stderr:
@@ -155,7 +154,6 @@ else:
                 
                 execution = sandbox.run_code(code_to_run)
                 
-                # FIXED: Properly join list of strings
                 if execution.logs.stdout:
                     log_str = "".join(execution.logs.stdout)
                     output_log += f"📄 **STDOUT**:\n```\n{log_str}\n```"
@@ -202,7 +200,7 @@ CRITICAL INSTRUCTIONS:
     elif mode == "Visualization Mode":
         return base_prompt + "\n\nMODE: VISUALIZATION\n- Focus on matplotlib/seaborn.\n- Create visually striking plots.\n- NO INTERNET ACCESS."
     elif mode == "Web Mode":
-        return base_prompt + "\n\nMODE: WEB ARCHITECT\n- Focus on modern web technologies (Flask, HTML, CSS).\n- **YOU HAVE INTERNET ACCESS**. You can use `requests` to fetch real data.\n- Structure code for web servers or API clients."
+        return base_prompt + "\n\nMODE: WEB ARCHITECT\n- Focus on modern web technologies (Flask, HTML, CSS).\n- **YOU HAVE INTERNET ACCESS**. Use `requests` to fetch real data.\n- Structure code for web servers or API clients."
     elif mode == "C++ Mode":
         return base_prompt + "\n\nMODE: C++ DEVELOPER\n- Focus on modern C++ (17/20).\n- Use iostream, vector, string.\n- Ensure code compiles with g++.\n- NO INTERNET ACCESS."
     else: # Code Mode
@@ -215,7 +213,8 @@ def parse_code_update(text, current_project, active_file):
     """
     clean_text = clean_thought_process(text)
     
-    # Matches ```lang ... ```
+    # Robust Regex for code blocks (Python, C++, JS, HTML, CSS)
+    # Using multi-line string construction to prevent truncation errors
     pattern = (
         r"```"          # Opening backticks
         r"(?:python|cpp|c\+\+|javascript|html|css)?"  # Optional language identifier
@@ -247,11 +246,14 @@ def process_query(user_input, session_data, project_state, active_file, mode):
         yield history, project_state, project_state[active_file], session_data
         return
 
+    # Update Internal State
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": f"🧠 *Fyodor is thinking in {mode}...*"})
     
+    # Yield UI Update
     yield history, project_state, project_state[active_file], session_data
 
+    # Context Construction
     sys_prompt = get_system_prompt(mode)
     
     project_context = "\n\n=== CURRENT PROJECT FILES ===\n"
@@ -260,6 +262,7 @@ def process_query(user_input, session_data, project_state, active_file, mode):
     
     messages = [{"role": "system", "content": sys_prompt + project_context}]
 
+    # Build Chat History for API
     for msg in history[:-2][-8:]:
         if isinstance(msg, dict):
             if msg['role'] == 'user':
@@ -312,7 +315,6 @@ def manual_run_project(project_state, session_data, mode):
     if session_data is None: session_data = init_session()
     history = session_data["history"]
     history.append({"role": "user", "content": "▶️ **Run Project (Manual)**"})
-    # PASS MODE TO EXECUTOR
     result = execute_code_e2b(project_state, mode=mode)
     history.append({"role": "assistant", "content": f"**Execution Result:**\n\n{result}"})
     return history, session_data
@@ -324,7 +326,7 @@ def run_code_review(code, session_data):
     history.append({"role": "assistant", "content": "🤔 *Reviewing...*"})
     
     messages = [
-        {"role": "system", "content": "You are a Senior Architect. Review the code for security and efficiency."},
+        {"role": "system", "content": "Senior Architect Code Review. Check for security and efficiency."},
         {"role": "user", "content": code}
     ]
     try:
@@ -336,6 +338,11 @@ def run_code_review(code, session_data):
         history[-1]["content"] = f"❌ Review Failed: {e}"
     return history, session_data
 
+def clear_session_chat(session_data):
+    if session_data is None: session_data = init_session()
+    session_data["history"] = []
+    return [], session_data
+
 def update_active_file(selected_file, project_data):
     lang = detect_language(selected_file)
     return gr.update(value=project_data.get(selected_file, ""), language=lang)
@@ -379,10 +386,11 @@ body, .gradio-container { background-color: var(--bg-dark) !important; color: va
 .chatbot .bot.message { background-color: transparent !important; border-left: 2px solid #333 !important; }
 textarea, input[type="text"] { background-color: #3c3c3c !important; border: 1px solid #3c3c3c !important; color: #fff !important; border-radius: 2px !important; }
 textarea:focus, input[type="text"]:focus { border-color: var(--accent) !important; box-shadow: 0 0 0 1px var(--accent) !important; }
-button { border-radius: 2px !important; text-transform: uppercase; font-weight: 600; letter-spacing: 0.5px; font-size: 12px !important; }
+button { border-radius: 2px !important; text-transform: uppercase; font-weight: 600; letter-spacing: 0.5px; font-size: 12px !important; transition: all 0.2s ease-in-out !important; }
 button.primary { background: var(--accent) !important; color: #fff !important; border: none !important; }
+button.primary:hover { background-color: #4DA2E5 !important; box-shadow: 0 0 8px rgba(58, 150, 221, 0.4); }
 button.secondary { background: #333 !important; color: #fff !important; border: 1px solid #444 !important; }
-button:hover { filter: brightness(1.1); }
+button.secondary:hover { background-color: #444 !important; border-color: #555 !important; }
 .cm-editor { border: 1px solid var(--border-col) !important; border-radius: 0 !important; }
 #top-bar { background-color: var(--sidebar-bg); border-bottom: 1px solid var(--border-col); min-height: 50px; padding: 0 10px !important; display: flex; align-items: center; justify-content: space-between; }
 h1 { font-size: 1.2rem !important; color: var(--text-main) !important; font-weight: normal !important; margin: 0 !important; padding: 0 !important; background-color: transparent !important; border: none !important; display: flex; align-items: center; }
@@ -404,14 +412,16 @@ def run_app():
         sidebar_visible = gr.State(True)
 
         with gr.Column(elem_classes="main-panel"):
+            # Top Bar
             with gr.Row(elem_id="top-bar"):
                 with gr.Row(scale=1): gr.Markdown("# FYODOR IDE // PRO")
                 with gr.Row(scale=1):
                     sidebar_toggle_btn = gr.Button("👁️ Toggle Sidebar", size="sm", variant="secondary")
                     timer_display = gr.Textbox(value="READY", show_label=False, interactive=False, container=False, elem_classes="status-label")
 
+            # Workspace
             with gr.Row():
-                # LEFT: EDITOR
+                # Editor (Left)
                 with gr.Column(scale=4):
                     with gr.Row(elem_classes="file-tabs"):
                         file_dropdown = gr.Dropdown(choices=["main.py"], value="main.py", show_label=False, container=False, scale=2)
@@ -425,13 +435,14 @@ def run_app():
                         run_manual_btn = gr.Button("▶ EXECUTE SCRIPT", variant="primary")
                         review_btn = gr.Button("🕵 REVIEW CODE", size="sm", variant="secondary")
 
-                # CENTER: TERMINAL
+                # Terminal (Right)
                 with gr.Column(scale=2):
                     mode_radio = gr.Radio(choices=["Code Mode", "C++ Mode", "Data Mode", "Web Mode"], value="Code Mode", label="SYSTEM MODE", elem_classes="mode-radio")
                     chatbot = gr.Chatbot(label="TERMINAL", height=600, render_markdown=True, elem_classes="chatbot", show_label=False)
                     with gr.Row():
                         msg_input = gr.Textbox(placeholder="Type command...", show_label=False, scale=4, container=False)
                         run_btn = gr.Button("RUN", variant="primary", scale=1)
+                        clear_btn = gr.Button("🗑️", variant="secondary", scale=0, min_width=40)
 
                 # RIGHT: SIDEBAR
                 with gr.Column(scale=1, min_width=250, visible=True) as sidebar_col:
@@ -465,17 +476,16 @@ def run_app():
         run_btn.click(process_query, [msg_input, session_state, project_state, active_file_state, mode_radio], [chatbot, project_state, code_editor, session_state])
 
         file_dropdown.change(update_active_file, [file_dropdown, project_state], [code_editor]).then(lambda f: f, [file_dropdown], [active_file_state])
-        
         code_editor.change(save_file_change, [code_editor, active_file_state, project_state], [project_state])
         save_btn.click(manual_save_file, [code_editor, active_file_state, project_state], [project_state])
-        
-        # New File: Updates Dropdown, Project State, and Code Editor (Value+Lang)
         new_file_btn.click(create_new_file, [new_file_txt, project_state], [file_dropdown, project_state, code_editor])
 
-        # PASS MODE RADIO TO MANUAL RUN
         run_manual_btn.click(manual_run_project, [project_state, session_state, mode_radio], [chatbot, session_state])
         review_btn.click(run_code_review, [code_editor, session_state], [chatbot, session_state])
         
+        clear_btn.click(clear_session_chat, [session_state], [chatbot, session_state])
+        
+        # Mini Agent: DICT FORMAT (Gradio 6)
         mini_input.submit(lambda m, h: (h or []) + [{"role": "user", "content": m}, {"role": "assistant", "content": robust_search(m)}], [mini_input, mini_chat], [mini_chat])
         mini_btn.click(lambda m, h: (h or []) + [{"role": "user", "content": m}, {"role": "assistant", "content": robust_search(m)}], [mini_input, mini_chat], [mini_chat])