Update app.py

app.py

@@ -96,6 +96,58 @@ def remove_file():
     logging.info("All uploaded files removed")
     return redirect(url_for('index'))
 
+from flask import jsonify
+
+@app.route('/download_and_store', methods=['GET'])
+def download_and_store():
+    file_url = request.args.get('file_url')
+    
+    if not file_url:
+        response = {
+            'status': 'error',
+            'message': 'No file URL provided'
+        }
+        logging.warning("No file URL provided")
+        return jsonify(response), 400
+    
+    try:
+        # Fetch the file from the provided URL
+        response = requests.get(file_url)
+        if response.status_code == 200:
+            # Extract filename from the URL
+            output_filename = os.path.basename(file_url)
+            output_path = os.path.join(UPLOAD_FOLDER, output_filename)
+            
+            # Save the file to the output folder
+            with open(output_path, 'wb') as f:
+                f.write(response.content)
+            
+            logging.info(f"File {output_filename} downloaded and saved to {output_path}")
+            
+            # JSON response with file details
+            response_data = {
+                'status': 'success',
+                'file_name': output_filename,
+                'file_path': output_path,
+                'message': f'File {output_filename} downloaded and stored successfully'
+            }
+            return jsonify(response_data), 200
+        else:
+            response_data = {
+                'status': 'error',
+                'message': f"Failed to download file from {file_url}. Status code: {response.status_code}"
+            }
+            logging.error(f"Failed to download file from {file_url}. Status code: {response.status_code}")
+            return jsonify(response_data), 400
+    
+    except Exception as e:
+        logging.error(f"Error downloading file from {file_url}: {e}")
+        response_data = {
+            'status': 'error',
+            'message': f"Error downloading file from {file_url}: {str(e)}"
+        }
+        return jsonify(response_data), 500
+
 @app.route('/process', methods=['POST'])
 def process_file():
     uploaded_files = session.get('uploaded_files', [])