Update app.py

app.py

@@ -13,23 +13,64 @@ if uploaded_file and method != "Select a method":
     if method == "Raw Markdown":
         st.markdown(markdown_text)
 
+    # elif method == "Boxing Method":
+    #     def parse_markdown_into_boxes(text):
+    #         pattern = r'(^#+ .*$)'
+    #         parts = re.split(pattern, text, flags=re.MULTILINE)
+    #         sections = []
+    #         for i in range(1, len(parts), 2):
+    #             header = parts[i].strip()
+    #             content = parts[i + 1].strip() if i + 1 < len(parts) else ""
+    #             sections.append({"header": header, "content": content})
+    #         return sections
+
+    #     st.subheader("📦 Formatted Output")
+    #     for section in parse_markdown_into_boxes(markdown_text):
+    #         with st.container(border=True):
+    #             if section["header"]:
+    #                 st.markdown(f"#### {section['header']}")
+    #             st.markdown(section["content"])
     elif method == "Boxing Method":
         def parse_markdown_into_boxes(text):
-            pattern = r'(^#+ .*$)'
-            parts = re.split(pattern, text, flags=re.MULTILINE)
+            # Remove ```markdown block if it exists
+            cleaned = text.strip()
+            if cleaned.startswith("```markdown"):
+                cleaned = cleaned[len("```markdown"):].strip()
+            if cleaned.endswith("```"):
+                cleaned = cleaned[:-3].strip()
+    
+            # Split based on Markdown headers (## Box or ## Summary)
+            pattern = r'(^## .*$)'
+            parts = re.split(pattern, cleaned, flags=re.MULTILINE)
+    
             sections = []
             for i in range(1, len(parts), 2):
                 header = parts[i].strip()
                 content = parts[i + 1].strip() if i + 1 < len(parts) else ""
                 sections.append({"header": header, "content": content})
             return sections
-
-        st.subheader("📦 Formatted Output")
-        for section in parse_markdown_into_boxes(markdown_text):
-            with st.container(border=True):
-                if section["header"]:
-                    st.markdown(f"#### {section['header']}")
+    
+        st.subheader("📦 Boxing Method Output")
+    
+        all_sections = parse_markdown_into_boxes(markdown_text)
+        boxes = [s for s in all_sections if not s["header"].lower().startswith("## summary")]
+        summary = next((s for s in all_sections if s["header"].lower().startswith("## summary")), None)
+    
+        # Display boxes in two columns
+        col1, col2 = st.columns(2)
+        for i, section in enumerate(boxes):
+            target_col = col1 if i % 2 == 0 else col2
+            with target_col.container(border=True):
+                st.markdown(f"#### {section['header']}")
                 st.markdown(section["content"])
+    
+        # Display summary below full width
+        if summary:
+            st.markdown("---")
+            with st.container(border=True):
+                st.markdown(f"### {summary['header']}")
+                st.markdown(summary['content'])
+
 
     # elif method == "Outline Method":
     #     st.subheader("📝 Outline Method Output")