Update app.py

app.py

@@ -376,4 +376,23 @@ async def get_annotators():
     # Convert the Counter to a list of dictionaries
     annotators = [{"annotator_id": annotator_id, "prompt_count": count} for annotator_id, count in annotator_counts.items()]
 
-    return JSONResponse(content=annotators)
+    return JSONResponse(content=annotators)
+@app.get("/prompt_by_annotator/{story_id}/{annotator_id}")
+async def get_prompt_by_annotator(story_id: str, annotator_id: str):
+    print(f"Fetching prompt for story_id: {story_id}, annotator_id: {annotator_id}")
+    prompt = prompts_collection.find_one({"story_id": story_id, "annotator_id": int(annotator_id)})
+    print(f"Prompt retrieved: {prompt}")  # Debugging output
+    if prompt:
+        return {"story_id": story_id, "prompt": prompt.get("prompt", "")}
+    else:
+        return {"story_id": story_id, "prompt": ""}
+
+@app.get("/summary_by_annotator/{story_id}/{annotator_id}")
+async def get_summary_by_annotator(story_id: str, annotator_id: str):
+    print(f"Fetching summary for story_id: {story_id}, annotator_id: {annotator_id}")
+    summary = summaries_collection.find_one({"story_id": story_id, "annotator_id": int(annotator_id)})
+    print(f"Summary retrieved: {summary}")  # Debugging output
+    if summary:
+        return {"story_id": story_id, "summary": summary.get("summary", "")}
+    else:
+        return {"story_id": story_id, "summary": ""}