1 files changed, 7 insertions, 1 deletions
diff --git a/app.py b/app.py
index 8fa15de..d1183e0 100644
--- a/app.py
+++ b/app.py
@@ -12,7 +12,13 @@ def hello_world():
 
 @app.route('/badge/<package>.svg')
 def badge(package: str):
-    return render_template('badge.svg.jinja', versions=repos.get_versions(package))
+    versions = repos.get_versions(package)
+    newest_version = max(v for family in versions.values() for v in family.values())
+    print(newest_version)
+    rendered = render_template('badge.svg.jinja', versions=versions, newest_version=newest_version)
+    response = make_response(rendered)
+    response.headers['Content-Type'] = 'image/svg+xml'
+    return response
 
 
 if __name__ == '__main__':