Fixed bug + added CORS

Author: KacemMathlouthi

app.py

@@ -4,6 +4,7 @@
 from src.config.settings import get_settings
 from src.logger.logger import get_logger
 from src.routers import router_generator 
+from fastapi.middleware.cors import CORSMiddleware
 
 settings = get_settings()
 logger = get_logger(__file__)
@@ -22,6 +23,16 @@
 app = FastAPI(
     title="AI Report Generator API App SMARTSHIELD",
 )
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins, you can also specify a list of allowed origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all HTTP methods (GET, POST, etc.)
+    allow_headers=["*"],  # Allows all headers
+)
+
 logger.info(f"Starting App : \n {ascii_art}")
 
 logger.info("App Ready")
@@ -37,7 +48,7 @@ async def root():
         uvicorn.run(
             app,
             port=8002,
-            host="localhost",
+            host="0.0.0.0",
 
         )
     except KeyboardInterrupt as ki : 

src/routers/router_generator.py

@@ -29,16 +29,14 @@ async def generate_report(generate_report_request: GenerateReportRequest):
         with open(report_pdf_file_path, "rb") as report_file:
             pdf_content = io.BytesIO(report_file.read())
 
-        os.remove(report_pdf_file_path)
-
         # Return the PDF content as a StreamingResponse
         pdf_content.seek(0)
         return StreamingResponse(
             pdf_content,
             media_type="application/pdf",
             headers={"Content-Disposition": "attachment; filename=report.pdf"}
         )
-
+    
     except Exception as e:
         logger.error(f"Critical Error occurred in router_generator.generate_report: {e}")
         raise HTTPException(status_code=500, detail="An error occurred while generating the report")

src/services/service_generator.py

@@ -2,6 +2,7 @@
 from crewai import Crew
 import uuid
 import os
+from fastapi.responses import FileResponse
 from typing import Dict 
 from src.config.settings import get_settings
 from src.logger.logger import get_logger
@@ -30,13 +31,18 @@ async def agenerate_report(threat : str , threat_data : Dict ) :
         )
         for task in crew.tasks:
             result = task.execute()
-            
-        pdf_file_path = "cybersecurity_report" + str(uuid.uuid4()) + ".pdf" 
+
+        path_prefix = r"C:\\Users\\"
+        pdf_file_name = r"cybersecurity_report" + str(uuid.uuid4()) + ".pdf" 
+        pdf_file_path = path_prefix + pdf_file_name
+        print("PDF file path :", pdf_file_path)
         html_report = generate_html_report(result, client)
-                
+        print(html_report)
         html_to_pdf(html_report, pdf_file_path)
+        print("HTML to PDF conversion done")
+
+        return pdf_file_path
 
-        return pdf_file_path 
 
     except Exception as e :
         logger.error(f"Error occured in service_report_generator.generate_report : {e}")