add flask server to render images

README.md

@@ -286,7 +286,7 @@ fig = module.generate_confidence_curves(results, confidence_config)
 fig.show()
 ```
 
-![Alt text](assets/example_confidence_curves.png)
+![Confidence Curve](assets/example_confidence_curves.png)
 
 
 ## Further References

app.py

@@ -1,6 +1,11 @@
-from pathlib import Path
+import os
+import re
 import sys
+import threading
+from pathlib import Path
+
 import gradio as gr
+from flask import Flask, abort, send_from_directory
 
 
 def read_markdown(filepath):
@@ -16,17 +21,52 @@ def read_markdown(filepath):
                     end_index = i
                     break
         if end_index is None:
-            return ''.join(
-                lines)  # If no end delimiter found, return entire content
+            return ''.join(lines)  # If no end delimiter found, return entire content
         else:
-            return ''.join(lines[end_index +
-                                 1:])  # Return content after end delimiter
-
+            return ''.join(lines[end_index + 1:])  # Return content after end delimiter
 
 local_path = Path(sys.path[0])
 filepath = local_path / "README.md"
 markdown_content = read_markdown(filepath)
 
+# Debug: Print the markdown content to check if it includes the image syntax
+print("Markdown Content:")
+print(markdown_content)
+
+app = Flask(__name__)
+
+@app.route('/assets/<path:filename>')
+def serve_assets(filename):
+    # Safeguard against path traversal attacks and invalid paths
+    safe_path = os.path.normpath(filename)
+    if safe_path.startswith("..") or os.path.isabs(safe_path):
+        print(f"Invalid path: {filename}")
+        abort(400)  # Bad request for invalid paths
+
+    asset_path = local_path / 'assets' / safe_path
+    if not asset_path.is_file():
+        print(f"File not found: {asset_path}")
+        abort(404)  # File not found
+
+    try:
+        return send_from_directory(local_path / 'assets', safe_path)
+    except Exception as e:
+        print(f"Error serving file: {e}")
+        abort(403)  # Forbidden
+
+def run_flask():
+    # Use '0.0.0.0' to ensure the server is accessible from any network interface
+    app.run(port=7861, host='0.0.0.0')
+
+flask_thread = threading.Thread(target=run_flask)
+flask_thread.start()
+
+# Replace this with your Hugging Face Space's external URL
+external_url = "https://sea-ai-det-metrics.hf.space/"  # Update this to match your space's URL
+
+# Update the image paths in the markdown to use the external Flask server URL
+markdown_content = re.sub(r'assets/([^)\s]+)', rf'{external_url}/assets/\1', markdown_content)
+
 with gr.Blocks() as demo:
     gr.Markdown(markdown_content)
 

requirements.txt

@@ -1,4 +1,5 @@
 git+https://github.com/huggingface/evaluate@main
 git+https://github.com/SEA-AI/seametrics@develop
 fiftyone
-plotly
+plotly
+flask