Update app.py

app.py

@@ -10,7 +10,11 @@ from htrflow.volume.volume import Collection
 from htrflow.pipeline.pipeline import Pipeline
 
 DEFAULT_OUTPUT = "alto"
-CHOICES = ["txt", "alto", "page", "json"]
+FORMAT_CHOICES = ["letter_english", "letter_swedish", "spread_english", "spread_swedish"]
+FILE_CHOICES = ["txt", "alto", "page", "json"]
+
+FormatChoices = Literal["letter_english", "letter_swedish", "spread_english", "spread_swedish"]
+FileChoices = Literal["txt", "alto", "page", "json"]
 
 PIPELINE_CONFIGS = {
     "letter_english": {
@@ -116,7 +120,7 @@ PIPELINE_CONFIGS = {
 }
 
 @spaces.GPU
-def htrflow_htr_url(image_path: str, document_type: Literal["letter_english", "letter_swedish", "spread_english", "spread_swedish"] = "letter_swedish", output_format: Literal["txt", "alto", "page", "json"] = DEFAULT_OUTPUT, custom_settings: Optional[str] = None) -> str:
+def htrflow_htr_url(image_path: str, document_type: FormatChoices = "letter_swedish", output_format: FileChoices = DEFAULT_OUTPUT, custom_settings: Optional[str] = None, server_name= "https://gabriel-htrflow-mcp.hf.space/") -> str:
     """
     Process handwritten text recognition (HTR) on uploaded images and return both file content and download link.
     
@@ -152,7 +156,7 @@ def htrflow_htr_url(image_path: str, document_type: Literal["letter_english", "l
         str: JSON string containing both the file content and download link:
              {
                  "content": "file_content_here",
-                 "file_path": "[file_name](http://your-server:port/gradio_api//file=/tmp/gradio/{temp_folder}/{file_name}.{file_format})"
+                 "file_path": "[file_name](https://gabriel-htrflow-mcp.hf.space/gradio_api/file=/tmp/{temp-folder}/{file-name}.{file-format})"
              }
     """
     if not image_path:
@@ -198,9 +202,10 @@ def htrflow_htr_url(image_path: str, document_type: Literal["letter_english", "l
                 file_content = f.read()
 
             
+            
             file_name = Path(output_file_path).name
-            temp_folder = Path(output_file_path).parent.name
-            markdown_link = f"[{file_name}](https://gabriel-htrflow-mcp.hf.space/gradio_api/file=/tmp/gradio/{temp_folder}/{file_name})"
+            download_url = f"{server_name}/gradio_api/file={output_file_path}"
+            markdown_link = f"[{file_name}]({download_url})"
             
             result = {
                 "content": file_content,
@@ -231,8 +236,8 @@ def create_htrflow_mcp_server():
         fn=htrflow_htr_url,
         inputs=[
             gr.Image(type="filepath", label="Upload Image or Enter URL"),
-            gr.Dropdown(choices=["letter_english", "letter_swedish", "spread_english", "spread_swedish"], value="letter_swedish", label="Document Type"),
-            gr.Dropdown(choices=CHOICES, value=DEFAULT_OUTPUT, label="Output Format"),
+            gr.Dropdown(choices=FORMAT_CHOICES, value="letter_swedish", label="Document Type"),
+            gr.Dropdown(choices=FILE_CHOICES, value=DEFAULT_OUTPUT, label="Output Format"),
             gr.Textbox(label="Custom Settings (JSON)", placeholder="Optional custom pipeline settings", value=""),
         ],
         outputs=gr.Textbox(label="HTR Result (JSON)", lines=10),