Update flask_Character.py

flask_Character.py

@@ -358,7 +358,7 @@ def _generate_response_internal(
     if not email_text:
         print(f"[{datetime.now()}] _generate_response_internal: Email text is empty.")
         return "Cannot generate reply for empty email text."
-    
+
     try:
         llm = ChatGroq(model="meta-llama/llama-4-scout-17b-16e-instruct", temperature=0.7, max_tokens=800, groq_api_key=api_key)
         prompt_template_str="""
@@ -581,7 +581,7 @@ async def extract_email_data(request: ProcessEmailRequest):
     """
     if extracted_emails_collection is None:
         raise HTTPException(status_code=503, detail="MongoDB not available.")
-    
+
     current_date = date.today() # Get current date for context
 
     print(f"[{datetime.now()}] /extract-data: Received request.")
@@ -589,20 +589,35 @@ async def extract_email_data(request: ProcessEmailRequest):
         print(f"[{datetime.now()}] /extract-data: Calling internal processing function.")
         # Run blocking LLM call in a thread pool
         extracted_data = await asyncio.to_thread(_process_email_internal, request.email_text, request.groq_api_key, current_date)
-        
+
         print(f"[{datetime.now()}] /extract-data: Internal processing complete. Preparing for DB insert.")
         # Convert Pydantic model to dictionary for MongoDB insert, handling _id alias
         # Use model_dump for Pydantic v2
         data_to_insert = extracted_data.model_dump(by_alias=True, exclude_none=True, exclude={'id'})
 
-        print(f"[{datetime.now()}] /extract-data: Inserting into MongoDB...")
+        # --- NEW CONVERSION FOR MONGODB ---
+        # MongoDB's BSON doesn't natively support Python's datetime.date type.
+        # It expects datetime.datetime. Convert all date fields to datetime.datetime.
+        if 'appointments' in data_to_insert:
+            for appt in data_to_insert['appointments']:
+                if isinstance(appt.get('start_date'), date):
+                    appt['start_date'] = datetime.combine(appt['start_date'], datetime.min.time())
+                if isinstance(appt.get('end_date'), date) and appt.get('end_date') is not None:
+                    appt['end_date'] = datetime.combine(appt['end_date'], datetime.min.time())
+        if 'tasks' in data_to_insert:
+            for task_item in data_to_insert['tasks']:
+                if isinstance(task_item.get('due_date'), date):
+                    task_item['due_date'] = datetime.combine(task_item['due_date'], datetime.min.time())
+        # --- END NEW CONVERSION ---
+
+        print(f"[{datetime.now()}] /extract-data: Inserting into MongoDB... Data: {data_to_insert}") # Add data logging
         # Use asyncio.to_thread for blocking MongoDB insert operation
         insert_result = await asyncio.to_thread(extracted_emails_collection.insert_one, data_to_insert)
-        
+
         # Update the extracted_data object with the MongoDB-generated ID
         extracted_data.id = str(insert_result.inserted_id)
         print(f"[{datetime.now()}] /extract-data: Data inserted into MongoDB. ID: {extracted_data.id}")
-        
+
         return extracted_data
     except ValueError as ve:
         raise HTTPException(status_code=400, detail=str(ve))
@@ -645,7 +660,7 @@ async def query_extracted_emails(query_params: ExtractedEmailQuery = Depends()):
         cursor = await asyncio.to_thread(extracted_emails_collection.find, mongo_query)
         # Use to_list to limit results and convert to list
         results = await asyncio.to_thread(lambda: list(cursor.limit(query_params.limit)))
-        
+
         # Convert MongoDB documents to ExtractedData Pydantic models
         return [ExtractedData(**doc) for doc in results]
     except Exception as e:
@@ -683,7 +698,7 @@ async def query_generated_replies(query_params: GeneratedReplyQuery = Depends())
         cursor = await asyncio.to_thread(generated_replies_collection.find, mongo_query)
         # Use to_list to limit results and convert to list
         results = await asyncio.to_thread(lambda: list(cursor.limit(query_params.limit)))
-        
+
         # Convert MongoDB documents to GeneratedReplyData Pydantic models
         return [GeneratedReplyData(**doc) for doc in results]
     except Exception as e: