rajkhanke commited on
Commit
2ee470b
·
verified ·
1 Parent(s): e41c65d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from flask import Flask, render_template, request, jsonify, Response
2
  import requests
3
  from bs4 import BeautifulSoup
 
4
 
5
  app = Flask(__name__)
6
 
@@ -142,17 +143,20 @@ def fetch_weeks():
142
 
143
  @app.route('/proxy-image')
144
  def proxy_image():
145
- # Get the external URL from the query parameter
146
  external_url = request.args.get('url')
147
  if not external_url:
148
  return "Missing URL", 400
149
 
150
  try:
151
- # Fetch the image from the external server
152
- resp = requests.get(external_url, timeout=10)
153
- return Response(resp.content, mimetype=resp.headers.get('Content-Type', 'image/jpeg'))
 
 
 
154
  except Exception as e:
155
  return str(e), 500
 
156
 
157
  if __name__ == '__main__':
158
  app.run(debug=True)
 
1
  from flask import Flask, render_template, request, jsonify, Response
2
  import requests
3
  from bs4 import BeautifulSoup
4
+ from flask import stream_with_context
5
 
6
  app = Flask(__name__)
7
 
 
143
 
144
  @app.route('/proxy-image')
145
  def proxy_image():
 
146
  external_url = request.args.get('url')
147
  if not external_url:
148
  return "Missing URL", 400
149
 
150
  try:
151
+ # Use streaming so that the response is sent in chunks
152
+ resp = requests.get(external_url, timeout=10, stream=True)
153
+ return Response(
154
+ stream_with_context(resp.iter_content(chunk_size=1024)),
155
+ mimetype=resp.headers.get('Content-Type', 'image/jpeg')
156
+ )
157
  except Exception as e:
158
  return str(e), 500
159
+
160
 
161
  if __name__ == '__main__':
162
  app.run(debug=True)