Update report_generator.py

report_generator.py

@@ -4,7 +4,7 @@ from fpdf import FPDF
 import matplotlib.pyplot as plt
 
 def generate_csv_report(filename="report.csv"):
-    """Generate a simple CSV report and return the filename."""
+    """Generate a sample CSV report and return its filename."""
     data = [
         ["Name", "Age", "City"],
         ["Alice", 30, "New York"],
@@ -17,7 +17,7 @@ def generate_csv_report(filename="report.csv"):
     return filename
 
 def generate_xlsx_report(filename="report.xlsx"):
-    """Generate a simple XLSX report using openpyxl and return the filename."""
+    """Generate a sample XLSX report using openpyxl and return its filename."""
     wb = Workbook()
     ws = wb.active
     data = [
@@ -31,27 +31,31 @@ def generate_xlsx_report(filename="report.xlsx"):
     wb.save(filename)
     return filename
 
-def generate_pdf_report(filename="report.pdf"):
-    """Generate a simple PDF report using fpdf and return the filename."""
+def generate_pdf_report(filename="report.pdf", subject=None):
+    """Generate a PDF report. If a subject is provided, it is used as the report title and context."""
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
-    lines = [
-        "Name: Alice, Age: 30, City: New York",
-        "Name: Bob, Age: 25, City: Los Angeles",
-        "Name: Charlie, Age: 35, City: Chicago"
-    ]
-    for line in lines:
-        pdf.cell(200, 10, txt=line, ln=True)
+    if subject:
+        pdf.cell(200, 10, txt=f"Report on: {subject}", ln=True)
+        pdf.ln(10)
+        pdf.multi_cell(0, 10, txt=f"This is an auto-generated report for the request: '{subject}'.\n\n"
+                                   "Data and analysis would be included here based on the subject.")
+    else:
+        lines = [
+            "Name: Alice, Age: 30, City: New York",
+            "Name: Bob, Age: 25, City: Los Angeles",
+            "Name: Charlie, Age: 35, City: Chicago"
+        ]
+        for line in lines:
+            pdf.cell(200, 10, txt=line, ln=True)
     pdf.output(filename)
     return filename
 
 def generate_diagram_report(filename="diagram.png"):
-    """Generate a sample diagram using matplotlib and return the filename."""
-    # Sample data for a bar chart.
+    """Generate a sample diagram using matplotlib and return its filename."""
     labels = ["A", "B", "C", "D"]
     values = [10, 24, 36, 18]
-    
     plt.figure(figsize=(6, 4))
     plt.bar(labels, values, color='skyblue')
     plt.title("Sample Diagram")