Spaces:

10gen
/

deepsearchitv2

Runtime error

App Files Files Community

Guiyom commited on Mar 12, 2025

Commit

09a8e4b

verified ·

1 Parent(s): d2319df

Update app.py

Browse files

Files changed (1) hide show

app.py +33 -9

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def capture_visual_screenshot(srcdoc: str) -> str:
     driver = None
     try:
         driver = webdriver.Chrome(options=options)
-        driver.set_window_size(1080, 720)  # Adjust per your expected visual dimensions
         # Write the srcdoc to a temporary HTML file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmp_file:
@@ -47,7 +47,7 @@ def capture_visual_screenshot(srcdoc: str) -> str:
             file_url = "file://" + tmp_file.name
         driver.get(file_url)
-        time.sleep(3)  # Allow time for JavaScript (e.g., mermaid) to render
         screenshot_png = driver.get_screenshot_as_png()
         image = Image.open(BytesIO(screenshot_png))
@@ -2215,12 +2215,10 @@ def replace_focus_placeholders(report_html: str, context: str, initial_query: st
             focus_html = generate_focus_snippet(instructions, context, initial_query, crumbs)
             # Remove any outer HTML or body tags from the generated snippet:
             focus_html = re.sub(r"<\/?(html|head|body)[^>]*>", "", focus_html, flags=re.DOTALL|re.IGNORECASE).strip()
-            # Now wrap in a single container with reduced font size:
             return (
                 f'<!-- Focus {placeholder_num} Start -->'
-                f'<p style="text-align: center;">-----------------------------------------------------------------------</p>'
-                f'{focus_html}'
-                f'<p style="text-align: center;">-----------------------------------------------------------------------</p>'
                 f'<!-- Focus {placeholder_num} End -->'
             )
         except Exception as e:
@@ -3013,10 +3011,36 @@ class ReportGenerator:
         pre, div {{ white-space: pre-wrap; }}
         ol, ul {{ font-size: 11pt; margin-left: 20px; line-height: 1.5; }}
         hr {{ border: 1px solid #ccc; margin: 20px 0; }}
-        table {{ border-collapse: collapse; width: 100%; margin-bottom: 10px; }}
-        th, td {{ border: 1px solid #ccc; padding: 8px; text-align: left; }}
-        th {{ background-color: #f2f2f2; }}
         .page-break {{ page-break-before: always; }}
     </style>
   </head>
   <body>

     driver = None
     try:
         driver = webdriver.Chrome(options=options)
+        driver.set_window_size(1200, 800)  # Adjust per your expected visual dimensions
         # Write the srcdoc to a temporary HTML file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmp_file:
             file_url = "file://" + tmp_file.name
         driver.get(file_url)
+        time.sleep(5)  # Allow time for JavaScript (e.g., mermaid) to render
         screenshot_png = driver.get_screenshot_as_png()
         image = Image.open(BytesIO(screenshot_png))
             focus_html = generate_focus_snippet(instructions, context, initial_query, crumbs)
             # Remove any outer HTML or body tags from the generated snippet:
             focus_html = re.sub(r"<\/?(html|head|body)[^>]*>", "", focus_html, flags=re.DOTALL|re.IGNORECASE).strip()
+            # Wrap the entire focus content in one div with a dedicated CSS class
             return (
                 f'<!-- Focus {placeholder_num} Start -->'
+                f'<div class="focus-placeholder">{focus_html}</div>'
                 f'<!-- Focus {placeholder_num} End -->'
             )
         except Exception as e:
         pre, div {{ white-space: pre-wrap; }}
         ol, ul {{ font-size: 11pt; margin-left: 20px; line-height: 1.5; }}
         hr {{ border: 1px solid #ccc; margin: 20px 0; }}
+        table {{
+             border-collapse: collapse;
+             width: 100%;
+             margin-bottom: 10px;
+             table-layout: fixed; /* force fixed layout so columns wrap */
+        }}
+        th, td {{
+             border: 1px solid #ccc;
+             padding: 8px;
+             text-align: left;
+             word-wrap: break-word; /* allow word wrapping */
+             overflow-wrap: break-word;
+        }}
+        th {{
+             background-color: #f2f2f2;
+        }}
         .page-break {{ page-break-before: always; }}
+        .focus-placeholder {{
+             border: 1px solid black;
+             padding: 10px;
+             font-size: 12pt;
+             line-height: 1.6;
+             margin-bottom: 10px;
+             word-wrap: break-word;
+        }}
+        .focus-placeholder p {{
+             margin: 0;
+             padding: 0;
+             border: none;
+        }}
     </style>
   </head>
   <body>