Update app.py

app.py

@@ -19,6 +19,9 @@ REGION_MODEL_PATH = "Kansallisarkisto/court-records-region-detection"
 TROCR_PROCESSOR_PATH = "Kansallisarkisto/multicentury-htr-model-onnx"
 TROCR_MODEL_PATH = "Kansallisarkisto/multicentury-htr-model-onnx"
 
+# Allowed source paths for input images
+ALLOWED_SOURCES = ('https://astia.narc.fi', '/tmp/gradio')
+
 login(token=os.getenv("HF_TOKEN"), add_to_git_credential=True)
 
 print(f"Is CUDA available: {torch.cuda.is_available()}")
@@ -85,12 +88,10 @@ def is_allowed_source(file_path):
         """
         Filter function to determine if a file source is allowed.
         """
-        # Allowed paths for input images
-        allowed_sources = ('https://astia.narc.fi', '/tmp/gradio')
         # Check allowed paths
-        if file_path.startswith(allowed_sources):
+        if file_path.startswith(ALLOWED_SOURCES):
             return True
-        print(f"File path not explicitly allowed: {file_path}")
+        print(f"File path not allowed: {file_path}")
         return False  
     
 async def get_filepath(request):
@@ -158,12 +159,12 @@ with gr.Blocks(theme=gr.themes.Monochrome(), title="Multicentury HTR Demo") as d
         
     async def run_pipeline(image, request: gr.Request):
         if request:
-            print("=== Request Information ===")
-            print(f"Request URL: {request.url}")
-            print(f"Request method: {request.method}")
-            print(f"Client host: {request.client.host}")
-            print(f"Headers: {dict(request.headers)}")
-            print(f"Query params: {dict(request.query_params)}")
+            #print("=== Request Information ===")
+            #print(f"Request URL: {request.url}")
+            #print(f"Request method: {request.method}")
+            #print(f"Client host: {request.client.host}")
+            #print(f"Headers: {dict(request.headers)}")
+            #print(f"Query params: {dict(request.query_params)}")
             file_path = await get_filepath(request)
             # Only files from allowed sources are processed
             if not is_allowed_source(file_path):