Update app.py

app.py

@@ -713,21 +713,28 @@ Now provide the result directly.
     # When creating the iframe, properly escape single quotes in the srcdoc attribute
     result = result.strip().strip("```").strip()
     
-    # Replace single quotes with their HTML entity to avoid breaking the srcdoc attribute
-    escaped_result = result.replace("'", "'")
+    # Properly escape the HTML content for srcdoc attribute
+    # Replace single quotes with HTML entities to avoid breaking the srcdoc attribute
+    escaped_content = result.replace("'", "'").replace('"', """)
     
-    # Create the iframe with proper styling and dimensions
-    htmloutput = f"""<div class="graph-container" style="width:100%; overflow:auto;">
-    <iframe class="visual-frame" width="1400" height="800" 
-            style="width:100%; min-height:600px; max-height:800px; border:none;" 
-            srcdoc="{escaped_result}"></iframe>
-    </div>"""
+    # Create a responsive container with proper width constraints
+    htmloutput = f"""
+    <div class="d3-visualization" style="width:100%; max-width:1400px; margin:0 auto; overflow:hidden;">
+        <iframe 
+            class="visual-frame" 
+            width="100%" 
+            height="600" 
+            style="border:none; width:100%; min-height:600px;"
+            srcdoc="{escaped_content}">
+        </iframe>
+    </div>
+    """
     
-    logging.info(f"The code produced for this graph placeholder:\n{placeholder_text}\n\n")
+    logging.info(f"The code produced for this graph placeholder:\n{placeholder_text}")
     return htmloutput
 
 def replace_graph_placeholders(report_html: str, context: str, initial_query: str, crumbs: str) -> str:
-    pattern = r"\[\[Graph Placeholder (\d+):(.*?)\]\]"  # Capture placeholder number
+    pattern = r"\[\[Graph Placeholder (\d+):(.*?)\]\]"
     
     def placeholder_replacer(match):
         placeholder_num = match.group(1)
@@ -736,11 +743,11 @@ def replace_graph_placeholders(report_html: str, context: str, initial_query: st
         
         try:
             visual_html = generate_graph_snippet(instructions, context, initial_query, crumbs)
-            # Add a container div for better styling control
-            return f'<div class="graph-wrapper">\n<!-- Graph {placeholder_num} Start -->\n{visual_html}\n<!-- Graph {placeholder_num} End -->\n</div>'
+            return f'<!-- Graph {placeholder_num} Start -->\n<div class="graph-container" style="margin:20px 0;">\n{visual_html}\n</div>\n<!-- Graph {placeholder_num} End -->'
         except Exception as e:
             logging.error(f"Graph {placeholder_num} failed: {str(e)}")
-            return f'<div class="error-message"><!-- ERROR GENERATING GRAPH {placeholder_num} --></div>'
+            error_msg = f'<div class="error-message" style="color:red; border:1px solid red; padding:10px; margin:10px 0;">Error generating Graph {placeholder_num}: {str(e)}</div>'
+            return error_msg
 
     return re.sub(pattern, placeholder_replacer, report_html, flags=re.DOTALL)