feat: add support for images

gradio_app.py

@@ -1,21 +1,24 @@
 import gradio as gr
 from deepengineer.backend.gradio_tools import run_agent_stream
+from deepengineer.common_path import DATA_DIR
 
 with gr.Blocks() as demo:
     gr.Markdown("# Agent Interface with Real‑Time Tool Logging")
     user_input  = gr.Textbox(label="User Message")
+    
+    log_output = gr.Textbox(label="Tool Invocation Log", interactive=False)
+    
     agent_output = gr.Markdown(
         label="Agent Response", 
     )
-    log_output = gr.Textbox(label="Tool Invocation Log", interactive=False)
 
     send = gr.Button("Send")
     send.click(
         fn=run_agent_stream,
         inputs=[user_input],
-        outputs=[log_output, agent_output],
+        outputs=[agent_output, log_output],
         concurrency_limit=2,
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(allowed_paths=[DATA_DIR])

src/deepengineer/backend/gradio_tools.py

@@ -10,45 +10,20 @@ from deepengineer.common_path import DATA_DIR
 
 def parse_markdown_images(markdown_text: str, image_dir: Path) -> str:
     """
-    Parse markdown text and convert image references to absolute file paths
-    that Gradio can display.
-    
-    Args:
-        markdown_text: The markdown text containing image references
-        image_dir: The directory containing the generated images
-    
-    Returns:
-        Modified markdown text with absolute file paths for images
+    Replace image references ![Title](image_name.png) with ![Title](image_dir/image_name.png)
     """
     if not markdown_text or not image_dir:
         return markdown_text
-    
-    # Pattern to match markdown image syntax: ![alt text](image_path)
+
     image_pattern = r'!\[([^\]]*)\]\(([^)]+)\)'
-    
+
     def replace_image_path(match):
         alt_text = match.group(1)
-        image_path = match.group(2)
-        
-        # If the path is already absolute, return as is
-        if Path(image_path).is_absolute():
-            return f'![{alt_text}]({image_path})'
-        
-        # If it's a relative path, try to find it in the image directory
-        if image_dir.exists():
-            # Look for the image file in the image directory
-            for img_file in image_dir.glob('*'):
-                if img_file.name == Path(image_path).name:
-                    return f'![{alt_text}]({img_file.absolute()})'
-            
-            # If not found by name, try to find any image file
-            image_files = list(image_dir.glob('*.png')) + list(image_dir.glob('*.jpg')) + list(image_dir.glob('*.jpeg'))
-            if image_files:
-                return f'![{alt_text}]({image_files[0].absolute()})'
-        
-        # If no image found, return the original reference
-        return match.group(0)
-    
+        image_name = match.group(2)
+        # Always use image_dir/image_name
+        new_path = str(Path(image_dir) / image_name)
+        return f'![{alt_text}]({new_path})'
+
     return re.sub(image_pattern, replace_image_path, markdown_text)
 
 
@@ -98,6 +73,10 @@ def run_agent_stream(user_input: str):
     
     if final_answer and image_dir:
         final_answer = parse_markdown_images(final_answer, image_dir)
+        
+    final_answer = final_answer.replace("```python", "")
+    final_answer = final_answer.replace("```markdown", "")
+    final_answer = final_answer.replace("```", "")
 
     # final yield: agent_output filled with processed markdown, log_output frozen
     yield (final_answer, log_buffer.rstrip())

src/deepengineer/deepsearch/main_agent.py

@@ -18,8 +18,8 @@ from pathlib import Path
 import queue
 
 
-def create_output_image_path():
-    random_name_images = random.randint(1000000, 9999999)
+def create_output_image_path(random_name_images: int | None = None):
+    random_name_images = random_name_images or random.randint(1000000, 9999999)
     output_image_path = Path(DATA_DIR) / f"images_{random_name_images}"
     output_image_path.mkdir(parents=True, exist_ok=True)
     return output_image_path

tests/backend/test_gradio_tools.py

@@ -0,0 +1,75 @@
+from pathlib import Path
+from deepengineer.backend.gradio_tools import parse_markdown_images
+from deepengineer.deepsearch.main_agent import create_output_image_path
+
+markdown = """
+# Nuclear Reactor: Structure and Components
+
+A nuclear reactor is a complex system designed to initiate and control a sustained nuclear chain reaction. The primary purpose of a 
+nuclear reactor is to generate heat through nuclear fission, which is then used to produce electricity. Below is a detailed explanation 
+of the main components and their functions, along with a simplified diagram.
+
+## Simplified Diagram of a Nuclear Reactor
+
+![Simplified Diagram of a Nuclear Reactor](nuclear_reactor_diagram.png)
+
+## Main Components and Their Functions
+
+### 1. Reactor Core
+The reactor core is the central part of the nuclear reactor where nuclear fission occurs. It contains:
+- **Nuclear Fuel**: Typically uranium oxide pellets arranged in fuel rods.
+- **Control Rods**: Made of materials like cadmium or boron, these rods absorb neutrons and regulate the fission rate.
+- **Moderator**: A material (often water) that slows down neutrons to sustain the chain reaction.
+
+### 2. Pressure Vessel
+A robust container that houses the reactor core and coolant under high pressure. It ensures the integrity and safety of the reactor core.
+
+### 3. Coolant
+A fluid (usually water) that circulates through the core to absorb the heat generated by fission. The heated coolant is then used to 
+produce steam.
+
+### 4. Steam Generator
+The heat from the coolant is transferred to a secondary water loop, which turns into steam. This steam is used to drive turbines.
+
+### 5. Turbine
+The steam produced in the steam generator is directed onto the blades of a turbine, causing it to spin. The turbine is connected to a 
+generator, which produces electricity.
+
+### 6. Cooling Tower
+Excess heat that is not converted into electricity is dissipated in the cooling tower. The cooling tower releases this heat into the 
+atmosphere.
+
+### 7. Containment Structure
+A steel-reinforced concrete dome or enclosure that isolates the reactor from the environment. It is designed to contain any radioactive 
+materials and prevent their release.
+
+## How a Nuclear Reactor Works
+
+1. **Nuclear Fission**: In the reactor core, uranium atoms undergo fission, releasing a significant amount of heat energy.
+2. **Heat Transfer**: The coolant absorbs this heat and transfers it to the steam generator.
+3. **Steam Production**: In the steam generator, water is converted into steam using the heat from the coolant.
+4. **Electricity Generation**: The steam drives the turbine, which is connected to a generator, producing electricity.
+5. **Cooling**: Excess heat is dissipated in the cooling tower, and the cycle repeats.
+
+## Sources
+
+- [World Nuclear Association - Nuclear Power 
+Reactors](https://world-nuclear.org/information-library/nuclear-fuel-cycle/nuclear-power-reactors/nuclear-power-reactors.aspx)
+- [Britannica - Nuclear Reactor](https://www.britannica.com/technology/nuclear-reactor)
+- [U.S. Department of Energy - How Does a Nuclear Reactor 
+Work?](https://www.energy.gov/ne/articles/nuclear-101-how-does-nuclear-reactor-work)
+- [Wikipedia - Nuclear Reactor](https://en.wikipedia.org/wiki/Nuclear_reactor)
+- [BYJU'S - Introduction to Nuclear Reactor](https://byjus.com/physics/nuclear-reactor-based-on-nuclear-fission/)
+
+This simplified diagram and explanation provide a basic understanding of the structure and function of a nuclear reactor.
+"""
+
+
+def test_parse_markdown_images():
+    image_dir = create_output_image_path(4351840)
+    real_image_path = image_dir / "nuclear_reactor_diagram.png"
+    output_image_path = "data/images_4351840/nuclear_reactor_diagram.png"
+    assert real_image_path.exists()
+    parsed_markdown = parse_markdown_images(markdown, image_dir)
+    assert output_image_path in parsed_markdown
+