inoaaer commited on
Commit
63237b9
·
1 Parent(s): 34ae053

Derive content-type from filename extension for reliable rendering

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -77,7 +77,7 @@ async def upload_file(
77
  file_id = uuid.uuid4().hex
78
  content = await file.read()
79
  filename = file.filename or "file"
80
- content_type = file.content_type or mimetypes.guess_type(filename)[0] or "application/octet-stream"
81
  safe_name = filename.replace(" ", "_")
82
 
83
  file_key = f"uploads/{file_id}/{safe_name}"
@@ -156,7 +156,8 @@ def serve_file(file_id: str):
156
  r2_key = meta["r2_key"]
157
  obj = s3.get_object(Bucket=R2_BUCKET, Key=r2_key)
158
  body = obj["Body"]
159
- content_type = meta.get("content_type", "application/octet-stream")
 
160
 
161
  def stream():
162
  while True:
 
77
  file_id = uuid.uuid4().hex
78
  content = await file.read()
79
  filename = file.filename or "file"
80
+ content_type = mimetypes.guess_type(filename)[0] or file.content_type or "application/octet-stream"
81
  safe_name = filename.replace(" ", "_")
82
 
83
  file_key = f"uploads/{file_id}/{safe_name}"
 
156
  r2_key = meta["r2_key"]
157
  obj = s3.get_object(Bucket=R2_BUCKET, Key=r2_key)
158
  body = obj["Body"]
159
+ filename = meta.get("filename", "")
160
+ content_type = mimetypes.guess_type(filename)[0] or meta.get("content_type") or "application/octet-stream"
161
 
162
  def stream():
163
  while True: