kn29 commited on
Commit
00dbf94
·
verified ·
1 Parent(s): 06e1d36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -3
app.py CHANGED
@@ -519,11 +519,26 @@ async def get_processing_results(session_id: str):
519
  )
520
  chunks_metadata = await chunks_cursor.to_list(length=None)
521
 
522
- # Clean up ObjectIds
 
523
  for chunk in chunks_metadata:
524
  chunk["_id"] = str(chunk["_id"])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
525
 
526
- # --- FIX: Convert potential datetime object to an ISO 8601 string ---
527
  processing_completed_at = session.get("processing_completed_at")
528
  if processing_completed_at:
529
  processing_completed_at = processing_completed_at.isoformat()
@@ -540,7 +555,7 @@ async def get_processing_results(session_id: str):
540
  "total_chunks": len(chunks_metadata),
541
  "chunks": chunks_metadata[:10] # Return first 10 chunks metadata
542
  },
543
- "processing_completed_at": processing_completed_at # Use the converted variable
544
  }
545
  )
546
 
 
519
  )
520
  chunks_metadata = await chunks_cursor.to_list(length=None)
521
 
522
+ # --- FIX: Convert datetime objects to ISO strings ---
523
+ # Clean up ObjectIds and datetime objects in chunks
524
  for chunk in chunks_metadata:
525
  chunk["_id"] = str(chunk["_id"])
526
+ if chunk.get("created_at"):
527
+ chunk["created_at"] = chunk["created_at"].isoformat()
528
+
529
+ # Clean up NER result datetime objects
530
+ if ner_result:
531
+ ner_result["_id"] = str(ner_result["_id"])
532
+ if ner_result.get("created_at"):
533
+ ner_result["created_at"] = ner_result["created_at"].isoformat()
534
+
535
+ # Clean up summary result datetime objects
536
+ if summary_result:
537
+ summary_result["_id"] = str(summary_result["_id"])
538
+ if summary_result.get("created_at"):
539
+ summary_result["created_at"] = summary_result["created_at"].isoformat()
540
 
541
+ # Convert session datetime objects
542
  processing_completed_at = session.get("processing_completed_at")
543
  if processing_completed_at:
544
  processing_completed_at = processing_completed_at.isoformat()
 
555
  "total_chunks": len(chunks_metadata),
556
  "chunks": chunks_metadata[:10] # Return first 10 chunks metadata
557
  },
558
+ "processing_completed_at": processing_completed_at
559
  }
560
  )
561