Update report_structure.py

report_structure.py

@@ -56,7 +56,6 @@ def generate_report(research_output: str, filename="research_report.pdf"):
     toc_h1_style = ParagraphStyle(
         'TOC_H1_Style',
         parent=styles['Heading2'],
-
     )
 
     # --- Title Page ---
@@ -71,8 +70,8 @@ def generate_report(research_output: str, filename="research_report.pdf"):
     # --- Table of Contents (Placeholder) ---
     story.append(Paragraph("Table of Contents", h1_style))
     toc_entries = extract_toc(research_output)
-    for level, title in toc_entries:
-         story.append(Paragraph(f"{title}", styles['Normal']))
+    for level, title_text in toc_entries:
+         story.append(Paragraph(f"{title_text}", styles['Normal']))
     story.append(PageBreak())
 
     # --- Main Content ---
@@ -84,11 +83,9 @@ def generate_report(research_output: str, filename="research_report.pdf"):
              story.append(Paragraph(para.strip(), normal_style))
              story.append(Spacer(1, 12))
 
-    doc.build(story, onFirstPage=add_page_number, onLaterPages=add_page_number) #<-Pass the function here
+    doc.build(story, onFirstPage=add_page_number, onLaterPages=add_page_number)
     return research_output
 
-
-
 def extract_title(text):
     """Extracts the title from the research output (simple heuristic)."""
     lines = text.split("\n")
@@ -103,12 +100,11 @@ def extract_toc(text):
     lines = text.split("\n")
     for i,line in enumerate(lines):
         if line.startswith("# ") and line.count(".")<=3:
-            toc.append((1, line[2:].strip()))  # Level 1 heading
-        elif line.startswith("## ")and line.count(".")<=3:
-            toc.append((2, line[3:].strip()))  # Level 2 heading
+            toc.append((1, line[2:].strip()))
+        elif line.startswith("## ") and line.count(".")<=3:
+            toc.append((2, line[3:].strip()))
     return toc
 
-
 def split_into_sections(text):
     """Splits the research output into sections based on headings."""
     sections = {}
@@ -128,10 +124,9 @@ def split_into_sections(text):
 
     return sections
 
-
 def add_page_number(canvas, doc):
     """Adds page numbers to the footer."""
     canvas.saveState()
     canvas.setFont('Times-Roman', 9)
-    canvas.drawString(letter[0] - 72, 0.75 * 72, "Page %d" % doc.page) #<-Added Closing parenthesis
-    canvas.restoreState()
+    canvas.drawString(letter[0] - 72, 0.75 * 72, "Page %d" % doc.page)
+    canvas.restoreState()