Update app.py

app.py

@@ -477,9 +477,27 @@ async def process_story(story: StoryCreationDTO, background_tasks: BackgroundTas
 
 #-----------------------Results Get End Point ______________________________________
 
+# @app.get("/tts/results/{task_id}")
+# async def get_results(task_id: str):
+
+#     if task_id not in tasks:
+#         return {"status": "not_found"}
+
+#     task = tasks[task_id]
+
+#     if task["status"] == "processing":
+#         return {"status": "processing"}
+
+#     if task["status"] == "failed":
+#         return {
+#             "status": "failed",
+#             "error": task["error"]
+#         }
+
+#     return task["result"]
+
 @app.get("/tts/results/{task_id}")
 async def get_results(task_id: str):
-
     if task_id not in tasks:
         return {"status": "not_found"}
 
@@ -491,10 +509,16 @@ async def get_results(task_id: str):
     if task["status"] == "failed":
         return {
             "status": "failed",
-            "error": task["error"]
+            "error": task.get("error", "Unknown error")
         }
 
-    return task["result"]
+    # Ensure result exists and has all required fields
+    result = task.get("result")
+    if result and all(k in result for k in ("fileName", "duration", "audioBase64")):
+        return {"status": "completed", **result}
+    else:
+        # If result is missing fields, mark as still processing
+        return {"status": "processing"}
 
 #----------------------------Test End Point to test tts inference------------------------------------