improve latex parsing

app.py

@@ -135,30 +135,30 @@ def update_inputs(task):
 def parse_latex_output(res):
     lines = res.split('\n')
     parsed_lines = []
-    in_tabular = False
+    in_latex = False
     
     for line in lines:
         line = line.strip()
         if not line:
             continue
         
-        if line.startswith('\\begin{tabular}') or line.startswith('\\end{tabular}'):
-            parsed_lines.append(f'$$ {line} $$')
-            in_tabular = line.startswith('\\begin{tabular}')
-            continue
+        latex_patterns = [r'\{', r'\}', r'\[', r'\]', r'\\', r'\$', r'_', r'^']
+        contains_latex = any(re.search(pattern, line) for pattern in latex_patterns)
         
-        if in_tabular:
-            parsed_lines.append(f'$$ {line} $$')
-            continue
+        is_key_value = ':' in line and not line.startswith('{') and not line.endswith('}')
         
-        if line.startswith('\\title') or line.startswith('\\author') or line.startswith('\\section'):
+        if contains_latex or is_key_value:
+            if not in_latex:
+                parsed_lines.append('$$')
+                in_latex = True
             parsed_lines.append(line)
-            continue
-        
-        if re.search(r'[\\{}$_^]', line) or any(keyword in line for keyword in ['\\hline', '\\begin', '\\end']):
-            parsed_lines.append(f'$$ {line} $$')
         else:
+            if in_latex:
+                parsed_lines.append('$$')
+                in_latex = False
             parsed_lines.append(line)
+    if in_latex:
+        parsed_lines.append('$$')
     
     return '\n'.join(parsed_lines)