packages

app/DocumentGeneration/live_pdf_generator.py

@@ -6,7 +6,13 @@ import io
 import tempfile
 import os
 from typing import Dict, Any, Optional, AsyncGenerator
-from fpdf2 import FPDF
+try:
+    from fpdf2 import FPDF
+except ImportError:
+    try:
+        from fpdf import FPDF
+    except ImportError:
+        raise ImportError("Neither fpdf2 nor fpdf package found. Please install fpdf2.")
 import base64
 
 import sys

app/DocumentGeneration/pdf_generator.py

@@ -39,7 +39,10 @@ def calculate_real_page_numbers(request: ExportRequest, filtered_sections):
     try:
         from fpdf2 import FPDF
     except ImportError:
-        raise Exception("fpdf2 is not installed. Please install it with: pip install fpdf2")
+        try:
+            from fpdf import FPDF
+        except ImportError:
+            raise Exception("Neither fpdf2 nor fpdf package found. Please install fpdf2.")
     
     # Create temporary PDF to calculate page numbers
     temp_pdf = FPDF()
@@ -185,7 +188,10 @@ def create_pdf_document(request: ExportRequest):
     try:
         from fpdf2 import FPDF
     except ImportError:
-        raise Exception("fpdf2 is not installed. Please install it with: pip install fpdf2")
+        try:
+            from fpdf import FPDF
+        except ImportError:
+            raise Exception("Neither fpdf2 nor fpdf package found. Please install fpdf2.")
     
     # Filter out empty sections to prevent blank pages
     try:

app/main.py

@@ -116,7 +116,6 @@ sys.path.append(os.path.join(os.path.dirname(__file__), 'CurrencyMapping'))
 
 from DocumentGeneration.pdf_generator import create_pdf_document
 from DocumentGeneration.word_generator import create_word_document
-from DocumentGeneration.live_pdf_generator import live_pdf_generator
 from DocumentGeneration.chart_generator import parse_embedded_chart_data, create_chart_from_data
 from DocumentGeneration.table_of_content import generate_table_of_contents_after_content, add_table_of_contents_page
 from DocumentGeneration.document_helpers import (
@@ -1600,73 +1599,73 @@ async def download_file(filename: str):
 # Live PDF Generation Endpoints
 # ──────────────────────────────────────────────────────────────────────────────
 
-@app.post("/live-pdf/start")
-async def start_live_pdf_session(request: Request, data: ExportRequest):
-    """Start a live PDF session that streams updates as content changes"""
-    try:
-        # Create a new live PDF session
-        session_id = await live_pdf_generator.create_live_session(
-            data.businessIdea or "default", 
-            data
-        )
-        
-        return {
-            "success": True,
-            "session_id": session_id,
-            "message": "Live PDF session started successfully"
-        }
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to start live PDF session: {str(e)}")
-
-@app.post("/live-pdf/update/{session_id}")
-async def update_live_pdf_session(session_id: str, data: ExportRequest):
-    """Update an existing live PDF session with new content"""
-    try:
-        success = await live_pdf_generator.update_session(session_id, data)
-        
-        if not success:
-            raise HTTPException(status_code=404, detail="Session not found")
-        
-        return {
-            "success": True,
-            "message": "PDF updated successfully"
-        }
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to update PDF: {str(e)}")
-
-@app.get("/live-pdf/stream/{session_id}")
-async def stream_live_pdf_updates(session_id: str):
-    """Stream live PDF updates using Server-Sent Events"""
-    async def event_generator():
-        async for chunk in live_pdf_generator.stream_pdf_updates(session_id):
-            yield chunk
-    
-    return StreamingResponse(
-        event_generator(),
-        media_type="text/event-stream",
-        headers={
-            "Cache-Control": "no-cache",
-            "Connection": "keep-alive",
-            "Access-Control-Allow-Origin": "*",
-            "Access-Control-Allow-Headers": "*"
-        }
-    )
-
-@app.delete("/live-pdf/close/{session_id}")
-async def close_live_pdf_session(session_id: str):
-    """Close and cleanup a live PDF session"""
-    try:
-        live_pdf_generator.close_session(session_id)
-        
-        return {
-            "success": True,
-            "message": "Live PDF session closed successfully"
-        }
-        
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to close session: {str(e)}")
+# @app.post("/live-pdf/start")
+# async def start_live_pdf_session(request: Request, data: ExportRequest):
+#     """Start a live PDF session that streams updates as content changes"""
+#     try:
+#         # Create a new live PDF session
+#         session_id = await live_pdf_generator.create_live_session(
+#             data.businessIdea or "default", 
+#             data
+#         )
+#         
+#         return {
+#             "success": True,
+#             "session_id": session_id,
+#             "message": "Live PDF session started successfully"
+#         }
+#         
+#     except Exception as e:
+#         raise HTTPException(status_code=500, detail=f"Failed to start live PDF session: {str(e)}")
+
+# @app.post("/live-pdf/update/{session_id}")
+# async def update_live_pdf_session(session_id: str, data: ExportRequest):
+#     """Update an existing live PDF session with new content"""
+#     try:
+#         success = await live_pdf_generator.update_session(session_id, data)
+#         
+#         if not success:
+#             raise HTTPException(status_code=404, detail="Session not found")
+#         
+#         return {
+#             "success": True,
+#             "message": "PDF updated successfully"
+#         }
+#         
+#     except Exception as e:
+#         raise HTTPException(status_code=500, detail=f"Failed to update PDF: {str(e)}")
+
+# @app.get("/live-pdf/stream/{session_id}")
+# async def stream_live_pdf_updates(session_id: str):
+#     """Stream live PDF updates using Server-Sent Events"""
+#     async def event_generator():
+#         async for chunk in live_pdf_generator.stream_pdf_updates(session_id):
+#             yield chunk
+#     
+#     return StreamingResponse(
+#         event_generator(),
+#         media_type="text/event-stream",
+#         headers={
+#             "Cache-Control": "no-cache",
+#             "Connection": "keep-alive",
+#             "Access-Control-Allow-Origin": "*",
+#             "Access-Control-Allow-Headers": "*"
+#         }
+#     )
+
+# @app.delete("/live-pdf/close/{session_id}")
+# async def close_live_pdf_session(session_id: str):
+#     """Close and cleanup a live PDF session"""
+#     try:
+#         live_pdf_generator.close_session(session_id)
+#         
+#         return {
+#             "success": True,
+#             "message": "Live PDF session closed successfully"
+#         }
+#         
+#     except Exception as e:
+#         raise HTTPException(status_code=500, detail=f"Failed to close session: {str(e)}")
 @app.get("/")
 def root():
     return {"status": "FastAPI is running 🚀"}

requirements.txt

@@ -12,7 +12,7 @@ supabase>=2.6.0
 asyncpg
 python-jose[cryptography]
 python-docx==0.8.11
-fpdf2
+fpdf2>=2.7.0
 Pillow==10.0.0
 requests
 PyJWT