fix content display of full documents

SlendGamer · Oct 1, 2024 · e4978b2 · e4978b2
1 parent 46c1afa
commit e4978b2
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/backend/app/ai.py b/backend/app/ai.py
@@ -22,7 +22,7 @@
 logger = logging.getLogger(__name__)
 
 
-async def generate_summary(file_path: str) -> str:
+async def generate_summary(file_path: str) -> dict[str, str | list[str | dict]]:
     try:
         with open(file_path, "r", encoding="utf-8") as f:
             text = f.read()
@@ -38,8 +38,7 @@ async def generate_summary(file_path: str) -> str:
             None, lambda: chat_llm.invoke(prompt)
         )
 
-        summary = summary_message.content
-
+        summary = {"summary": summary_message.content, "full_text": text}
         logger.info("Summary generated successfully.")
         return summary
 

diff --git a/backend/app/models.py b/backend/app/models.py
@@ -33,7 +33,7 @@ class Document(Base):
     user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
     title = Column(String(255), nullable=False, index=True)
     description = Column(Text, nullable=True)
-    # content = Column(Text)
+    content = Column(Text, nullable=True)
     file_path = Column(String(500), nullable=False)
     uploaded_at = Column(DateTime, default=datetime.utcnow)
     summary = Column(Text, nullable=True)

diff --git a/backend/app/routers/documents.py b/backend/app/routers/documents.py
@@ -24,6 +24,7 @@
 async def upload_document(
     title: str = Form(...),
     description: Optional[str] = Form(None),
+    content: Optional[str] = Form(None),
     tags: Optional[List[str]] = Form([]),
     file: UploadFile = File(...),
     db: Session = Depends(get_db),
@@ -36,6 +37,7 @@ async def upload_document(
         user_id=user_id,
         title=title,
         description=description,
+        content=content,
         file_path=file_path,
         uploaded_at=datetime.utcnow(),
     )
@@ -66,7 +68,8 @@ async def upload_document(
     embeddings = await generate_embeddings(file_path)
 
     # Update Document with summary
-    document.summary = summary
+    document.summary = summary["summary"]
+    document.content = summary["full_text"]
     db.commit()
     db.refresh(document)
 

diff --git a/backend/app/schemas.py b/backend/app/schemas.py
@@ -78,7 +78,7 @@ class DocumentDetailResponse(BaseModel):
     user_id: int
     title: str
     description: Optional[str]
-    # content: str
+    content: Optional[str]
     uploaded_at: datetime
     uploader: UserBase
     tags: List[str] = []