Update exporters.py

exporters.py

@@ -1,5 +1,5 @@
-import os, json, io
-from typing import Dict, Optional, List
+import os, json
+from typing import Dict
 from models import NarrativeData, CompanyInfo
 from jinja2 import Environment, FileSystemLoader, select_autoescape
 
@@ -48,12 +48,9 @@ def export_to_notion(outputs: Dict[str, str]) -> Dict[str, str]:
         return {"status": "skipped", "reason": "NOTION_API_KEY or NOTION_DATABASE_ID not set"}
 
     notion = Client(auth=api_key)
-    # Create one parent page, then child pages for each file
     parent = notion.pages.create(
         parent={"database_id": database_id},
-        properties={
-            "Name": {"title": [{"text": {"content": "PR/IR One-Pass Outputs"}}]}
-        }
+        properties={"Name": {"title": [{"text": {"content": "PR/IR One-Pass Outputs"}}]}},
     )
     parent_id = parent.get("id")
 
@@ -61,13 +58,13 @@ def export_to_notion(outputs: Dict[str, str]) -> Dict[str, str]:
         notion.pages.create(
             parent={"page_id": parent_id},
             properties={"Name": {"title": [{"text": {"content": name}}]}},
-            children=[
-                {"object": "block", "type": "paragraph", "paragraph": {"rich_text": [{"type": "text", "text": {"content": md}}]}}
-            ]
+            children=[{
+                "object": "block",
+                "type": "paragraph",
+                "paragraph": {"rich_text": [{"type": "text", "text": {"content": md}}]}
+            }]
         )
-    result["status"] = "ok"
-    result["parent_page_id"] = parent_id
-    return result
+    return {"status": "ok", "parent_page_id": parent_id}
 
 # ---------------- Google Docs Export (optional) ----------------
 def export_to_google_docs(outputs: Dict[str, str]) -> Dict[str, str]:
@@ -99,18 +96,18 @@ def export_to_google_docs(outputs: Dict[str, str]) -> Dict[str, str]:
         doc = docs_service.documents().create(body={"title": name}).execute()
         doc_id = doc.get("documentId")
 
-        # Simple insert of raw markdown (no conversion)
-        requests = [
-            {"insertText": {"location": {"index": 1}, "text": md}}
-        ]
+        requests = [{"insertText": {"location": {"index": 1}, "text": md}}]
         docs_service.documents().batchUpdate(documentId=doc_id, body={"requests": requests}).execute()
 
-        # Move to folder if provided
         if folder_id:
-            # first, get current parents
             file = drive_service.files().get(fileId=doc_id, fields="parents").execute()
             prev_parents = ",".join(file.get("parents", []))
-            drive_service.files().update(fileId=doc_id, addParents=folder_id, removeParents=prev_parents, fields="id, parents").execute()
+            drive_service.files().update(
+                fileId=doc_id,
+                addParents=folder_id,
+                removeParents=prev_parents,
+                fields="id, parents"
+            ).execute()
 
         result["docs"].append(f"https://docs.google.com/document/d/{doc_id}/edit")
     return result