Update main.py

main.py

@@ -91,21 +91,16 @@ def proxy(url):
 
     clean_url = full_url.replace('proxy/', '')
     clean_url = encode_url(clean_url)
-    should_abort = lambda: 'abort' in session and session['abort']
-
-    def generate(should_abort):
-        for chunk in req.iter_content(chunk_size=1024):
-            if should_abort():
-                break
-            yield chunk
     
     try:
-        start = time.time()
-
-        logging.info(f"Sending request to: {clean_url}")
         req = requests.get(clean_url, headers=headers, stream=True, timeout=2)
+        logging.info(f"TIME: {time.time()-start}, Status Code: {req.status_code}, Response Headers: {req.headers}")
+
+        end_time = time.time()  
+        elapsed_time = end_time - start_time  
+        print(f"\nTime taken for proxy: {elapsed_time} seconds\n")
 
-        return Response(generate(should_abort), content_type=req.headers['content-type'])
+        return Response(req.iter_content(chunk_size=1024), content_type=req.headers['content-type'])
 
     except Exception as e:
         logging.error(f"Error in proxy: {str(e)}")