Upload app.py

app.py

@@ -5,7 +5,7 @@ US Army Medical Research Papers Q&A
 
 import os
 import logging
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, Union
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI, HTTPException
@@ -79,13 +79,13 @@ class ImageSource(BaseModel):
     path: Optional[str]
     filename: Optional[str]
     score: Optional[float]
-    page: Optional[str] # could be int or str depending on metadata
+    page: Optional[Union[str, int]] # could be int or str depending on metadata
     file: Optional[str]
 
 class TextSource(BaseModel):
     text: str
     score: float
-    page: Optional[str]
+    page: Optional[Union[str, int]]
     file: Optional[str]
 
 class QueryResponse(BaseModel):