Update flask_Character.py

flask_Character.py

@@ -760,31 +760,21 @@ async def query_extracted_emails_endpoint(query_params: ExtractedEmailQuery = De
         raise HTTPException(status_code=500, detail=f"Error querying extracted emails: {e}")
 
 
-@app.get("/query-generated-replies", response_model=List[GeneratedReplyData], summary="Query generated replies from MongoDB")
+app.get("/query-generated-replies", response_model=List[GeneratedReplyData], summary="Query generated replies from MongoDB")
 async def query_generated_replies_endpoint(query_params: GeneratedReplyQuery = Depends()):
-    """
-    Queries stored generated email replies from MongoDB based on various filters.
-    """
     if generated_replies_collection is None:
-        raise HTTPException(status_code=503, detail="MongoDB not available for querying generated replies.")
-
+        raise HTTPException(status_code=503, detail="MongoDB not available for generated_replies.")
     mongo_query: Dict[str, Any] = {}
-    if query_params.language:
-        mongo_query["language"] = query_params.language
-    if query_params.style:
-        mongo_query["style"] = query_params.style
-    if query_params.tone:
-        mongo_query["tone"] = query_params.tone
+    if query_params.language: mongo_query["language"] = query_params.language
+    if query_params.style: mongo_query["style"] = query_params.style
+    if query_params.tone: mongo_query["tone"] = query_params.tone
 
     if query_params.from_date or query_params.to_date:
         date_query: Dict[str, datetime] = {}
-        if query_params.from_date:
-            date_query["$gte"] = datetime.combine(query_params.from_date, datetime.min.time())
-        if query_params.to_date:
-            date_query["$lt"] = datetime.combine(query_params.to_date + timedelta(days=1), datetime.min.time())
-        if date_query:
-            mongo_query["generated_at"] = date_query
-
+        if query_params.from_date: date_query["$gte"] = datetime.combine(query_params.from_date, datetime.min.time())
+        if query_params.to_date: date_query["$lt"] = datetime.combine(query_params.to_date + timedelta(days=1), datetime.min.time())
+        if date_query: mongo_query["generated_at"] = date_query
+        
     try:
         cursor = generated_replies_collection.find(mongo_query).sort("generated_at", -1).limit(query_params.limit)
         generated_docs_raw = await asyncio.to_thread(list, cursor)
@@ -796,4 +786,4 @@ async def query_generated_replies_endpoint(query_params: GeneratedReplyQuery = D
         return results
     except Exception as e:
         traceback.print_exc()
-        raise HTTPException(status_code=500, detail=f"Error querying generated replies: {e}")
+        raise HTTPException(status_code=500, detail=f"Error querying generated replies: {e}")