update

app.py

@@ -143,21 +143,11 @@ import re
 def modify_plan_of_sqls_html(html_content):
     soup = BeautifulSoup(html_content, 'html.parser')
 
-    # Remove the final table (verification result)
-    final_table = soup.find('table', id=lambda x: x and x.startswith('step-'))
-    if final_table:
-        final_table.decompose()
-
-    # Remove the Prediction line
+    # Remove only the Prediction line
     prediction_h3 = soup.find('h3', string=re.compile(r'Prediction:'))
     if prediction_h3:
         prediction_h3.decompose()
 
-    # Remove any extra <hr> tags at the end if present
-    last_hr = soup.find_all('hr')[-1]
-    if last_hr:
-        last_hr.decompose()
-
     return str(soup)