Update app.py

app.py

@@ -107,49 +107,96 @@ Do not name the company if name is not there and return just the report and noth
 
 
 def create_pdf_report(report_text):
-    
     """
-    Create a PDF from markdown text using the md-to-pdf API.
+    Create a PDF from markdown text using the md-to-pdf API with enhanced formatting.
     
     Args:
         report_text (str): Markdown formatted report text
         
     Returns:
         BytesIO: PDF file in memory buffer
-    """
     
+    Raises:
+        Exception: If PDF generation fails
+    """
     api_url = "https://md-to-pdf.fly.dev"
+    
+    # Enhanced CSS for better table formatting
     css = """
-h1, h2 {
-    color: MidnightBlue;
-}
-
-table {
-  border-collapse: collapse;
-}
-
-table, th, td {
-  border: 1px solid DimGray;
-}
-
-th, td {
-  text-align: left;
-  padding: 1em;
-}
-"""
+    @page {
+        margin: 20mm;
+        size: A4;
+    }
+    
+    body {
+        font-family: 'Helvetica', sans-serif;
+        line-height: 1.6;
+        color: #333;
+    }
+    
+    h1, h2 {
+        color: #191970; /* MidnightBlue */
+        border-bottom: 2px solid #eee;
+        padding-bottom: 0.3em;
+    }
+    
+    table {
+        width: 100%;
+        border-collapse: collapse;
+        margin: 1em 0;
+        page-break-inside: avoid;
+    }
+    
+    th {
+        background-color: #f8f9fa;
+        font-weight: bold;
+        text-align: left;
+    }
+    
+    th, td {
+        padding: 0.75em;
+        border: 1px solid #ddd;
+        word-wrap: break-word;
+    }
+    
+    tr:nth-child(even) {
+        background-color: #f9f9f9;
+    }
+    
+    .amount {
+        text-align: right;
+        white-space: nowrap;
+    }
+    """
+    
+    # Pre-process markdown to clean up table formatting
+    processed_md = "\n".join([line.rstrip() for line in report_text.split("\n")])
+    
     payload = {
-        'markdown': report_text,
+        'markdown': processed_md,
         'engine': 'weasyprint',
         'css': css
     }
     
-    response = requests.post(api_url, data=payload)
-    if response.status_code == 200:
-        # Return the PDF in a BytesIO buffer
-        from io import BytesIO
+    try:
+        response = requests.post(
+            api_url,
+            data=payload,
+            timeout=30  # Add timeout for API call
+        )
+        
+        response.raise_for_status()  # Raise exception for 4xx/5xx errors
+        
+        # Validate PDF content
+        if not response.content.startswith(b'%PDF-'):
+            raise ValueError("Invalid PDF content received from API")
+            
         return BytesIO(response.content)
-    else:
-        raise Exception(f"Failed to generate PDF: {response.status_code} - {response.text}")
+        
+    except requests.exceptions.RequestException as e:
+        raise Exception(f"API request failed: {str(e)}") from e
+    except ValueError as e:
+        raise Exception(f"Invalid PDF response: {str(e)}") from e
 
 def main():
     st.title("Quantitlytix AI")