Update app.py

app.py

@@ -408,12 +408,33 @@ async def process_story(story: StoryCreationDTO, background_tasks: BackgroundTas
 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.get("error", "Unknown error")}
-    return {"status": "completed", **task["result"]}
+        return {
+            "status": "failed",
+            "error": task.get("error", "Unknown error")
+        }
+
+    # Ensure result exists and has all required fields
+    result = task.get("result")
+    if result and all(k in result for k in ("fileName", "duration", "audioPath")):
+        #clearing cache
+        print(f"all fields are available {result}")
+        for file_path in download_cache.values():
+            if os.path.exists(file_path):
+                os.remove(file_path)
+        download_cache.clear()
+        
+        return {"status": "completed", **result}
+    else:
+        print(f"missing field {result}")
+        # If result is missing fields, mark as still processing
+        return {"status": "processing"}
 
 @app.get("/")
 def root():