try to fix embedded graph

app.py

@@ -419,22 +419,29 @@ def build_d3_graph():
         # Check if the HTML file exists
         html_file = Path("static/d3_dependency_graph.html")
         if html_file.exists():
-            # Read and modify the HTML content for direct embedding
+            # Read the HTML content
             html_content = html_file.read_text(encoding="utf-8")
             
-            # Wrap in a container with proper styling
+            # Extract only the body content (remove head tags that might cause issues)
+            body_match = re.search(r'<body.*?>(.*?)</body>', html_content, re.DOTALL)
+            if body_match:
+                body_content = body_match.group(1)
+            else:
+                body_content = html_content
+            
+            # Get the script content
+            script_match = re.search(r'<script.*?>(.*?)</script>', html_content, re.DOTALL)
+            script_content = script_match.group(1) if script_match else ""
+            
+            # Create a clean embedded version
             wrapped_html = f'''
-            <div id="d3-graph-container" style="
-                width: 100%; 
-                height: 600px; 
-                border: 1px solid #e2e8f0; 
-                border-radius: 8px; 
-                overflow: hidden;
-                background: white;
-                position: relative;
-            ">
-                {html_content}
+            <div id="d3-graph-container" style="width: 100%; height: 600px; border: 1px solid #e2e8f0; border-radius: 8px; overflow: hidden; background: white; position: relative;">
+                {body_content}
             </div>
+            <script src="https://d3js.org/d3.v7.min.js"></script>
+            <script>
+                {script_content}
+            </script>
             <style>
                 #d3-graph-container svg {{
                     width: 100% !important;
@@ -445,10 +452,14 @@ def build_d3_graph():
                     top: 10px !important;
                     left: 10px !important;
                     z-index: 1000 !important;
+                    background: rgba(255,255,255,0.9) !important;
+                    padding: 12px !important;
+                    border-radius: 6px !important;
+                    font-size: 14px !important;
                 }}
             </style>
             '''
-            gr.HTML(wrapped_html, elem_id="d3-dependency-graph")
+            gr.HTML(wrapped_html)
         else:
             gr.Markdown("⚠️ **D3 dependency graph not found.** The file `static/d3_dependency_graph.html` is missing.")